将csv文件中的几列转存为另一个csv文件
时间: 2023-06-17 22:03:35 浏览: 93
csv-merger:将保存csv文件的许多相同类型的数据合并到一个csv文件中。 所有输入的csv文件应具有相同的列数,否则将出现异常
您可以使用Python中的pandas库来读取原始csv文件,并将需要的几列存入新的csv文件中。以下是一个示例代码:
```python
import pandas as pd
# 读取原始csv文件
df = pd.read_csv('original.csv')
# 选择需要的几列
new_df = df[['column1', 'column2', 'column3']]
# 将新的数据保存为csv文件
new_df.to_csv('new.csv', index=False)
```
在代码中,您需要将`column1`、`column2`和`column3`替换为您需要的列名,并将`original.csv`替换为您的原始文件名,将`new.csv`替换为您要保存的新文件名。
此外,`index=False`用于防止pandas将默认的行标签保存到新文件中。
阅读全文