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)

以上， 问题解决。