使用python如何对一个csv文件写入新的一列,每个csv文件有三列,新的一列列名为标签,新的一列的内容为癌症
时间: 2024-09-09 10:04:35 浏览: 106
要在Python中对一个现有的CSV文件写入新的一列,你可以使用`csv`模块来实现这一功能。首先,你需要读取原始CSV文件,然后在适当的位置插入新列的数据,并将结果写入到一个新的CSV文件中。以下是一个简单的示例代码,展示了如何实现这一过程:
```python
import csv
# 假设原始CSV文件名为original.csv,新列的列名为"标签",新列的数据为"癌症"
original_csv = 'original.csv'
new_csv = 'modified.csv'
# 打开原始CSV文件,并创建一个csv.reader对象
with open(original_csv, mode='r', newline='', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
# 使用csv.writer对象写入新的CSV文件
with open(new_csv, mode='w', newline='', encoding='utf-8') as newfile:
writer = csv.writer(newfile)
for row in reader:
# 在每个行数据的末尾添加新列的数据
row.append('癌症')
writer.writerow(row)
# 这样就会在new_csv文件中每个原始行后面添加"癌症"这一列
```
这段代码首先读取名为`original.csv`的文件,然后为每一行数据添加一个新元素(在这个例子中是字符串"癌症"),最后将这些行写入到一个名为`modified.csv`的新文件中。请确保在运行这段代码前原始的CSV文件存在,并且有正确的路径。
阅读全文