以下是一个示例代码,展示了如何编写一个csv文件,检查列中的值,并写入额外的数据:
import csv
# 定义要写入csv文件的数据
data = [
['Name', 'Age', 'Country'],
['John', 25, 'USA'],
['Alice', 30, 'Canada'],
['Bob', 35, 'UK']
]
# 定义要检查的列的索引
column_index = 1
# 定义要写入的额外数据
extra_data = ['Young', 'Young', 'Middle-aged', 'Middle-aged']
# 打开csv文件并写入数据
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
# 打开csv文件并读取数据
with open('data.csv', 'r') as file:
reader = csv.reader(file)
rows = list(reader)
# 检查列中的值并写入额外的数据
for i in range(1, len(rows)):
value = rows[i][column_index]
rows[i].append(extra_data[i-1])
# 重新写入csv文件
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(rows)
print("CSV文件已更新")
在这个示例中,我们首先定义了要写入csv文件的数据。然后,我们定义了要检查的列的索引(从0开始计数),以及要写入的额外数据。
接下来,我们使用csv.writer
将数据写入csv文件。然后,我们使用csv.reader
读取csv文件的数据,并将其存储在一个列表中。
然后,我们遍历读取的数据,检查指定列的值,并将额外的数据添加到每一行的末尾。
最后,我们使用csv.writer
重新写入已经更新的数据到csv文件中。
运行以上代码后,你将会得到一个名为"data.csv"的csv文件,并且每个人的年龄后面都被添加了额外的数据。