提取csv文件的两列,以id为关键字,转存另一个csv文件
时间: 2023-05-17 15:05:04 浏览: 167
可以使用Python的pandas库来实现这个功能。首先,读取原始的csv文件,然后使用pandas的DataFrame对象来选择需要的两列,以id为关键字进行筛选,最后将结果保存到另一个csv文件中。以下是示例代码:
```python
import pandas as pd
# 读取原始csv文件
df = pd.read_csv('original.csv')
# 选择需要的两列,以id为关键字进行筛选
new_df = df[['id', 'column2']]
# 将结果保存到另一个csv文件中
new_df.to_csv('new.csv', index=False)
```
请注意,代码中的'original.csv'和'new.csv'需要替换为实际的文件名。
阅读全文