從 pandas 數據幀創建列表字典

Question

我正在嘗試基於 pandas dataframe 創建一個列表字典，我需要一個列表字典來傳遞給我的 Plotly 儀表板

In:
df.head()
Model    Make
Ford     F-150
Ford     Escape
Ford     Mustang
Jeep     Grand Cherokee
Jeep     Wrangler

我發現df.to_dict()按 header 列定向，但我需要根據相鄰行值定向。 唯一的方法是通過 Model 將我的 dataframe 重新塑造成列，並在它們各自的品牌之下？

Out:
makes_by_model= {
    'Ford': ['F-150', 'Escape', 'Mustang'],
    'Jeep': ['Wrangler', 'Grand Cherokee']
}

Answer 1

您可以groupby ，聚合到列表，返回一個帶有rec.array的to_records並從結果中構造一個字典：

dict(df.groupby('Model').agg(list).to_records())
# {'Ford': ['F-150', 'Escape', 'Mustang'], 'Jeep': ['Grand Cherokee', 'Wrangler']}