Pandas: startswith()函数实现拆分文件
一、需求
将一个CSV文件中按照 NUMBER 列分成两个 CSV 文件，一个文件中 NUMBER 列以 AB 开头，剩下的为另一个 CSV 文件。
二、实现代码
import pandas as pd data = pd.read_csv('example.csv',header = 0) df = pd.DataFrame(data) # print(df) df['bool'] = df['NUMBER'].str.startswith('AB') # 分成两部分 data1 = df[df['bool'] == True] data2 = df[df['bool'] == False] # 写成两个 CSV 文件 data1.to_csv('file1.csv',index = False) data2.to_csv('file2.csv',index = False)
以上， 问题解决。
