python实现按指定列拆分csv文件
时间: 2023-08-06 15:03:26 浏览: 59
csvkiller:按任何列分割 CSV 文件
如果要按照指定的列拆分csv文件,可以使用pandas库来实现。以下是一个简单的示例代码:
```python
import pandas as pd
# 读取csv文件
df = pd.read_csv('example.csv')
# 按指定列拆分
for col in ['col1', 'col3']:
# 获取指定列的数据
col_data = df[col]
# 创建新的csv文件并写入数据
col_data.to_csv(f'{col}.csv', index=False)
```
这段代码将会把名为`example.csv`的文件按照指定的列`col1`和`col3`拆分成两个文件,文件名分别为`col1.csv`和`col3.csv`,文件中仅包含对应列的数据,不包含列头。如果需要包含列头,只需在每个拆分的文件中添加`header=[col]`参数即可。
阅读全文