csv文档用python删除除制定列的其他列
时间: 2024-05-11 20:15:03 浏览: 43
Python删除n行后的其他行方法
可以使用 pandas 库中的 drop() 函数删除不需要的列,示例代码如下:
```
import pandas as pd
# 读取 csv 文件
df = pd.read_csv('file.csv')
# 删除需要保留的列以外的所有列
df = df.drop(df.columns.difference(['col1', 'col2']), axis=1)
# 将结果写入新的 csv 文件
df.to_csv('new_file.csv', index=False)
```
其中,`col1` 和 `col2` 分别为需要保留的列名,`axis=1` 表示按列删除,`index=False` 表示不将行索引写入新的 csv 文件中。
阅读全文