python实现拆分csv文件(无表头)
拆分依据:某一列的第一个字符或某几个字符
import glob
import os
import pandas as pd
def split_csv():
for dfile in glob.glob('2.csv'):
df = pd.read_csv(dfile, header=None)
for letter, group in df.groupby(df[0].str[0]):
group.to_csv((os.path.splitext(dfile)[0]) + '_{}.csv'.format(letter), index=False, header=False)
split_csv()
2.csv:
拆分之后:
2_A.csv
2_B.csv:
df[1]表示第几列,str[]表示拆分的依据哪几个字符。
此csv文件就是依据第二列的第2到5个字符(即年份)进行拆分的。