在多个CSV文件的末尾添加具有特定文件名的列和行

import os
import csv

# 指定CSV文件所在的目录
directory = 'path/to/your/csv/files'

# 遍历目录中的所有文件
for filename in os.listdir(directory):
    if filename.endswith('.csv'):
        # 构造新的文件名
        new_filename = f"{filename}_modified.csv"
        # 打开原始CSV文件进行读取
        with open(os.path.join(directory, filename), mode='r', newline='', encoding='utf-8') as infile:
            reader = csv.reader(infile)
            rows = list(reader)
        
        # 在每行的末尾添加文件名列
        for row in rows:
            row.append(filename)
        
        # 添加新的一行，包含文件名和其他信息
        rows.append([filename, 'additional info'])
        
        # 将修改后的内容写入新的CSV文件
        with open(os.path.join(directory, new_filename), mode='w', newline='', encoding='utf-8') as outfile:
            writer = csv.writer(outfile)
            writer.writerows(rows)

print("处理完成！")