使用groupby在熊猫中创建字典

时间:2018-08-14 19:14:02

标签: python pandas dictionary

我正在尝试为以下数据集使用字典:

id  value
1   a
1   b
1   c
2   e
2   f
2   g
3   h
3   g
3   l

输出应该是这样

{1: [a,b,c], 2:[e, f, g], 3: [h, g, l]} 

我知道一些有关如何创建字典的参考,但是都没有给出这样的输出。

谢谢。

1 个答案:

答案 0 :(得分:5)

Groupby和表单列表,然后创建字典

df.groupby('id')['value'].apply(list).to_dict()
# {1: ['a', 'b', 'c'], 2: ['e', 'f', 'g'], 3: ['h', 'g', 'l']}