Pandas: startswith()函数实现拆分文件

本文主要是介绍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)

以上， 问题解决。

这篇关于Pandas: startswith()函数实现拆分文件的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---