2020-12-04

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个字符（即年份）进行拆分的。