python - 用唯一的键和值列表从两列中创建一个字典

Question

我正在尝试创建一个具有唯一键和多个值的字典

DF :

key value
2   21
2   32
2   455
3   12
3   45
3   21

预期输出:

{'2' : ['21', '32', '455'], '3': ['12','45','21']}

代码 :

dict(zip(df['key'], df['value']))

需要一些帮助

Answer 1

第一个聚合列表 GroupBy.agg 然后转换 Series到字典 Series.to_dict :

d = df.groupby('key')['value'].agg(list).to_dict()
print (d)
{2: [21, 32, 455], 3: [12, 45, 21]}