用于在csv python中基于列名创建标题的For循环

import csv

# 假设我们有一个CSV文件，其内容如下：
# name,age,city
# Alice,30,New York
# Bob,25,Los Angeles

# 打开CSV文件
with open('example.csv', mode='r', newline='', encoding='utf-8') as csvfile:
    # 创建csv阅读器
    csvreader = csv.DictReader(csvfile)
    
    # 获取列名
    fieldnames = csvreader.fieldnames
    
    # 打印列名，检查是否正确读取
    print("列名:", fieldnames)
    
    # 如果需要，可以在这里修改列名
    # 例如，将'age'改为'年龄'
    fieldnames = [name if name != 'age' else '年龄' for name in fieldnames]
    
    # 创建一个新的CSV文件，使用修改后的列名作为标题
    with open('modified_example.csv', mode='w', newline='', encoding='utf-8') as new_csvfile:
        # 创建csv写入器
        csvwriter = csv.DictWriter(new_csvfile, fieldnames=fieldnames)
        
        # 写入标题行
        csvwriter.writeheader()
        
        # 写入数据行
        for row in csvreader:
            # 如果列名被修改，需要更新字典中的键
            updated_row = {new_name: row[old_name] for old_name, new_name in zip(fieldnames, fieldnames)}
            csvwriter.writerow(updated_row)

print("CSV文件已更新并保存为 'modified_example.csv'")