数据框到字典列表

Question

我有以下数据框：

key         category    amount
2019-01     abc         123
2019-01     xyz         111
2019-02     abc         87
2019-02     xyz         34
2019-03     abc         15
2019-04     def         64

我将如何将这些数据转换为下面的字典列表。本质上，每个 dict 代表一个类别以及该类别的所有日期和相应数量：

[
  {
    "category": "abc",
    "2019-01": 123,
    "2019-02": 87,
    "2019-03": 15
  },
  {
    "category": "xyz",
    "2019-01": 111,
    "2019-02": 34
  },
  {
    "category": "def,
    "2019-04": 64
  }
]

Answer 1

在 GroupBy.apply 中使用 lambda 函数并合并 2 个字典：

f = lambda x: {**{'category': x.name}, **dict(zip(x['key'], x['amount']))}
L = df.groupby('category', sort=False).apply(f).tolist()
print (L)
[{'category': 'abc', '2019-01': 123, '2019-02': 87, '2019-03': 15}, 
 {'category': 'xyz', '2019-01': 111, '2019-02': 34},
 {'category': 'def', '2019-04': 64}]