python修改csv文件的一个列名
时间: 2023-06-18 09:06:16 浏览: 141
要修改 CSV 文件中的一个列名,可以使用 Python 的 csv 模块。以下是一个简单的示例代码:
```python
import csv
# 打开 CSV 文件并读取数据
with open('example.csv', 'r') as csvfile:
reader = csv.DictReader(csvfile)
rows = list(reader)
# 修改列名
for row in rows:
row['old_col_name'] = row.pop('old_col_name')
row['new_col_name'] = row.pop('old_col_name')
# 写入新的 CSV 文件
with open('new_example.csv', 'w', newline='') as csvfile:
fieldnames = ['new_col_name', 'other_col_name1', 'other_col_name2']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
writer.writerows(rows)
```
在这个示例中,我们首先使用 csv 模块的 DictReader 类来读取 CSV 文件。然后,我们遍历每一行数据,并使用 pop 方法将原来的列名从行数据中删除,然后使用新的列名添加到行数据中。
最后,我们使用 DictWriter 类将修改后的数据写入新的 CSV 文件中。需要注意的是,我们在创建 DictWriter 对象时需要传入新的列名列表。如果 CSV 文件中还有其他列,也需要在列表中添加它们的列名。
阅读全文