以下是一个示例代码,演示如何按行合并多个CSV文件:
import csv
def merge_csv_files(input_files, output_file):
# 打开输出文件
with open(output_file, 'w', newline='') as f_out:
writer = csv.writer(f_out)
# 遍历输入文件列表
for file in input_files:
# 打开输入文件
with open(file, 'r') as f_in:
reader = csv.reader(f_in)
# 逐行写入输出文件
for row in reader:
writer.writerow(row)
# 输入文件列表
input_files = ['file1.csv', 'file2.csv', 'file3.csv']
# 输出文件名
output_file = 'merged_file.csv'
# 调用函数进行合并
merge_csv_files(input_files, output_file)
上述代码中,merge_csv_files
函数接受两个参数:input_files
和output_file
。input_files
是一个包含输入文件名的列表,output_file
是合并后输出的文件名。
函数内部使用csv.reader
读取输入文件的内容,并使用csv.writer
将内容写入输出文件。通过嵌套的with open
语句,可以确保在处理完每个输入文件后都正确关闭文件。
要使用这个示例代码,只需将输入文件的文件名和输出文件的文件名替换为实际的文件名,并将其保存为一个Python脚本,然后运行该脚本即可合并CSV文件。请确保所有要合并的CSV文件都与Python脚本在同一目录下。