python - pandas dataframe to_dict 两列作为索引，第三列作为值

Question

我有一个 Pandas 数据框如下:

User              ASIN             Rating
A23VKINWRY6J92    1476783284       5
A3HC4SRK7B2AXR    1496177029       5
AE12HJWB5ODOD     B00K2GAUC0       4
AL4RYO265J1G      061579615X       3

我想生成一个字典，其中有 2 列“User”和“ASIN”作为键，第三列“Rating”作为值。如下所示:

my_dict[A23VKINWRY6J92][1476783284] = 5
my_dict[A3HC4SRK7B2AXR][1496177029] = 5
my_dict[AE12HJWB5ODOD][B00K2GAUC0] = 4
my_dict[AL4RYO265J1G][061579615X] = 3

我该怎么做？

Answer 1

使用嵌套字典理解:

{u: {a: list(df.Rating[(df.User == u) & (df.ASIN == a)].unique()) for a in df.ASIN[df.User == u].unique()} for u in df.User.unique()}

请注意，这会映射到列表，因为结果值没有理由应该是唯一的。