>问题描述
在Pandas中,当我想按某一列的值,把一个原数据集分组,并将分组后的数据存为一个DataFrame(进一步可以存为CSV文件或其他),此时若该列(组)的属性值个数是未定的,该如何进行操作?
>问题解决
需要用到pandas的groupby方法:
http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.groupby.html
col_id = 'your_col_id' origin_data = pd.read_csv('./data/your_data.csv') gp = origin_data.groupby(col_id) index_list = list(gp.describe().index) for i in index_list: print 'now at group:', i origin_data.loc[origin_data[col_id] == i].to_csv('./data/divided/divided_data_'+str(i)+'.csv', index=False)