python生成csv文件

python生成csv文件

import csv
import random
import string
import time

# 记录开始时间
start_time = time.time()

# 设定文件名和数据量
num_rows = 100 * 10000 + 1
file_name = f'E:\\20240421_test_python\\data_{num_rows}.csv'

# 使用with语句自动关闭文件
with open(file_name, 'w', newline='', encoding='utf-8') as csvfile:
    # 创建一个csv写入器
    csvwriter = csv.writer(csvfile)

    # 写入列名
    csvwriter.writerow(['Random Integer', 'Random String'])

    # 生成10万个数据行
    for _ in range(num_rows):
        # 生成一个随机整数
        random_integer = random.randint(1, 1000000)

        # 生成一个长度为10的随机字符串
        random_string = ''.join(random.choices(string.ascii_uppercase + string.digits, k=10))

        # 写入一行数据
        csvwriter.writerow([random_integer, random_string])

print(f'CSV file with {num_rows} rows has been generated.')

# 记录结束时间
end_time = time.time()

# 计算并打印运行时间
elapsed_time = end_time - start_time
print(f'Elapsed time: {elapsed_time:.2f} seconds')