优化使用Python向CSV文件添加列可以通过以下步骤实现:
import csv
import tempfile
import shutil
def add_column_to_csv(input_file, output_file, column_name, column_data):
# 创建一个临时文件来保存修改后的数据
temp_file = tempfile.NamedTemporaryFile(mode='w', delete=False)
with open(input_file, 'r') as csv_in_file, temp_file:
reader = csv.DictReader(csv_in_file)
fieldnames = reader.fieldnames + [column_name]
writer = csv.DictWriter(temp_file, fieldnames)
writer.writeheader()
for row in reader:
row[column_name] = column_data
writer.writerow(row)
# 替换原始文件
shutil.move(temp_file.name, output_file)
这个函数接受四个参数:输入文件路径(input_file)、输出文件路径(output_file)、要添加的列名(column_name)以及列数据(column_data)。
add_column_to_csv('input.csv', 'output.csv', 'new_column', 'new_data')
这将向名为"input.csv"的文件添加一个名为"new_column"的列,列中的数据为"new_data"。结果将保存在名为"output.csv"的文件中。
该方法的优势是使用了Python内置的CSV模块,使得操作CSV文件更加简便和高效。它适用于各种需要向CSV文件添加列的场景,如数据处理、数据分析和生成报告等。
腾讯云相关产品推荐:云对象存储 COS(Cloud Object Storage)
希望这些信息能够帮助到您!
领取专属 10元无门槛券
手把手带您无忧上云