2014-01-24 57 views
3

我創建使用下面的語句我的數據的各個組:如何大熊貓組保存到不同的文件

df = pd.read_csv(file_path) 

grouped = df.groupby(df.some_parameter) 

我會那麼喜歡做的事情(在僞代碼):

for name, group in grouped: 
    'Some Text' + name = group 
    write to csv 

最終的結果是原始數據集的每個數據塊的一個單獨的.csv文件。

回答

7

你幾乎有

for name, group in grouped: 
    group.to_csv(path_to_disk) 
2

這個答案是非常有益的,我 - 感謝@mkln。

我只是想具體的東西添加到我自己的使用案例,其中涉及到有關文件命名的原點(「一些文本」 +名=組)。

您可以添加名稱和其他文本,例如當前日期,每個CSV文件名,所以我將創建一個函數返回當前的日期,然後用這個文件名。

因此:

from datetime import datetime 

def cur_date(): 
    return datetime.now().strftime("%Y-%m-%d") 

for name, group in grouped: 
    group.to_csv('{}_{}.csv'.format(name, cur_date()))