用随机生成的变量替换csv中的变量

import csv
import random
import string

def generate_random_string(length):
    return ''.join(random.choices(string.ascii_letters + string.digits, k=length))

def replace_variables_in_csv(input_file, output_file, columns_to_replace):
    with open(input_file, mode='r', newline='') as infile, open(output_file, mode='w', newline='') as outfile:
        reader = csv.DictReader(infile)
        fieldnames = reader.fieldnames
        writer = csv.DictWriter(outfile, fieldnames=fieldnames)
        writer.writeheader()
        
        for row in reader:
            for column in columns_to_replace:
                if column in row:
                    row[column] = generate_random_string(10)  # 生成一个长度为10的随机字符串
            writer.writerow(row)

# 示例用法
input_file = 'input.csv'
output_file = 'output.csv'
columns_to_replace = ['name', 'email']

replace_variables_in_csv(input_file, output_file, columns_to_replace)