python - 数据框到字典列表

Question

我有以下 df:

df = pd.DataFrame({"year":[2020,2020,2020,2021,2021,2021,2022,2022, 2022],"region":['europe','USA','africa','europe','USA','africa','europe','USA','africa'],'volume':[1,6,5,3,8,7,6,3,5]})

我希望将其转换为字典列表，以便在每个项目中仅提及一次年份。示例

[{'year':2020,'europe':1,'USA':6,'africa':5,}...]

当我这样做时:

df.set_index('year').to_dict('records')

我失去了岁月和名单

Answer 1

另一种在 to_dict(orient='records') 之前使用 pivot 的方法

df.pivot(
    index='year',
    columns='region',
    values='volume'
).reset_index().to_dict(orient='records')

#Output:
#[{'year': 2020, 'USA': 6, 'africa': 5, 'europe': 1},
# {'year': 2021, 'USA': 8, 'africa': 7, 'europe': 3},
# {'year': 2022, 'USA': 3, 'africa': 5, 'europe': 6}]