python - 使用 groupby 在 Pandas 中创建字典

Question

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

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]} 

我知道一些关于如何创建字典的引用资料，但没有一个提供这样的输出。

谢谢。

Answer 1

Groupby 和表单列表，然后创建一个字典

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