以下是一个示例代码,演示如何编写一个CSV文件,并给列名添加前缀:
import csv
def write_csv_with_prefix(filename, rows, prefix):
with open(filename, 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
# 获取列名
fieldnames = list(rows[0].keys())
# 给列名添加前缀
prefixed_fieldnames = [prefix + fieldname for fieldname in fieldnames]
# 写入表头
writer.writerow(prefixed_fieldnames)
# 写入数据
for row in rows:
writer.writerow(row.values())
# 示例数据
data = [
{'Name': 'John', 'Age': 25, 'City': 'New York'},
{'Name': 'Mary', 'Age': 30, 'City': 'San Francisco'},
{'Name': 'David', 'Age': 35, 'City': 'Los Angeles'}
]
# 调用函数,写入CSV文件
write_csv_with_prefix('data.csv', data, 'Prefix_')
这段代码定义了一个write_csv_with_prefix
函数,它接受一个文件名、数据行列表和前缀作为参数。函数首先打开文件,创建一个csv.writer
对象。然后,它从第一行数据中获取列名,并使用列表推导式给列名添加前缀。接下来,它使用writerow
方法写入表头,并循环写入每一行数据。在每一行中,我们使用row.values()
获取值的列表,并使用writerow
方法写入CSV文件。
最后,我们定义了一个示例数据列表,然后调用write_csv_with_prefix
函数将数据写入名为data.csv
的CSV文件中,并给列名添加了前缀Prefix_
。
上一篇:编写CouchDB的视图