数据帧到带有项列表的字典

Question

我有这个Dataframe:

df = 
key value
0    1   kpi1
1    1   kpi2
2    1   kpi3
3    2   kpi4
4    2   kpi5

我要创建一个字典，对于相同的键，有一个值列表，像这样:

d = {'1': ['kpi1', 'kpi2', 'kpi3'], '2': ['kpi4', 'kpi5']}

我试过使用df。我申请了，但找不到解决办法。你有什么主意吗?

Answer 1

试试这个:

df.groupby('key')['value'].apply(list).to_dict()

结果:

{1: ['kpi1', 'kpi2', 'kpi3'], 2: ['kpi4', 'kpi5']}