從一個 -python 創建多個數據幀

Question

我有一個大文件（> 500 行），列表中的每個唯一項目都有多個數據點，例如：

起司	重量	地點
豪達	1.4	鋁
豪達	2	德克薩斯州
豪達	1.2	加州
切達干酪	5.3	鋁
切達干酪	6	錳
查達爾	2	西澳
哈瓦蒂	4	加州
哈瓦蒂	4.2	鋁

我想為每個奶酪制作數據框來存儲相關數據

我有這個：

main_cheese_file = pd.read_csv('CheeseMaster.csv')

cut_the_cheese = main_cheese_file.cheese.unique()
melted = {elem: pd.DataFrame() for elem in cut_the_cheese}

for slice in melted.slice():
    melted[slice] = main_cheese_file[:][main_cheese_file.cheese == slice]

把它分成我想要的獨特的東西。

我想用它做的是制作可以為每個奶酪導出的df，奶酪名稱作為文件名。

到目前為止，我可以強迫它

 melted['Cheddar'].to_csv('Cheddar.csv') 

並得到切達干酪....

但我不想知道並在 500 行列表中輸入每種類型的奶酪......

有沒有辦法將此添加到我的循環中？

Answer 1

您可以通過 object 遍歷一個組

import pandas as pd

df = pd.read_csv('CheeseMaster.csv')
for k,v in df.groupby('cheese'):
    v.to_csv(f'{k}.csv', index=False)