數據框到帶有項目列表的字典

Question

我有這個數據框：

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.apply 但找不到解決方案。 你有什么想法嗎？

Answer 1

嘗試這個：

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

結果：

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