如何在每个csv文件中添加列，其中列的值取决于csv的名称

import csv
import os

# 获取指定目录下的所有CSV文件
csv_dir = "/path/to/csv/files"
csv_files = [file for file in os.listdir(csv_dir) if file.endswith(".csv")]

# 遍历每个CSV文件
for file in csv_files:
    csv_path = os.path.join(csv_dir, file)

    # 获取CSV文件的名称
    csv_name = file.split(".csv")[0]

    # 添加列到CSV文件中
    with open(csv_path, "r", newline="") as csv_file:
        reader = csv.reader(csv_file)
        rows = list(reader)

    # 添加新的列名到CSV文件的表头
    rows[0].append(csv_name)

    # 遍历CSV文件中的每一行数据
    for i in range(1, len(rows)):
        # 根据CSV文件名称赋予新列的值
        rows[i].append(csv_name)

    # 将更新后的数据写回到CSV文件中
    with open(csv_path, "w", newline="") as csv_file:
        writer = csv.writer(csv_file)
        writer.writerows(rows)