[Python][Pandas][ML学习笔记04]按(未定)列的属性值分类/筛选数据并保存为csv文件

>问题描述

在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)


猜你喜欢

转载自blog.csdn.net/shenpibaipao/article/details/80191302