按某一列的数据进行分组

最新推荐文章于 2024-06-30 18:41:59 发布

有梦想有行动

最新推荐文章于 2024-06-30 18:41:59 发布

阅读量1.8k

点赞数 2

分类专栏：数据预处理文章标签： python

本文链接：https://blog.csdn.net/CSDN_of_ding/article/details/108119007

版权

数据预处理专栏收录该内容

4 篇文章 1 订阅

订阅专栏

我的目标是把csv文件中的某一列按类别同时自动拆分文件，按照一个特定类别输出文件。

废话不多说上代码

import pandas as pd
#read data from csv
df=pd.read_csv('bilibili(2).csv',sep=',',engine='python',header=[1,2],encoding="UTF-8")
# get the columns of df
df.columns = ['title', 'type']
c = list(df.columns)

print(c[1:2])
grouped=df.groupby(c[1:2]) #according different categories to group by the df
#delete duplicated data
ind_frame=df.drop_duplicates(subset=['type'], keep='first')
print(ind_frame.type)
print(range(len(ind_frame)))
for name in ind_frame.type:
    print(name)
    df[df.type==name].to_csv("rawdata\\_"+u"%s" %name+".csv")