CSV文件导出脚本

import datetime

import csv
import os


# 获取当前时间
now = datetime.datetime.now()

# 格式化时间为年月日时分秒
formatted_time = now.strftime('%Y_%m_%d_%H%M%S')

# 拼接文件名
csv_filename = f'exported_data_{formatted_time}.csv'

print(csv_filename)

# 定义CSV文件的名称
csv_filename = csv_filename

# 定义CSV头部字段
csv_header = ['Column1', 'Column2', 'Column3']

# 定义CSV数据，这里是一个示例列表的列表
csv_data = [
    {'Column1': '值1', 'Column2': '值2', 'Column3': '值3'},
    {'Column1': '值4', 'Column2': '值5', 'Column3': '值6'}
]

# 确保当前目录存在，如果不存在则创建
if not os.path.exists(os.getcwd()):
    os.makedirs(os.getcwd())

# 打开文件并写入数据
with open(csv_filename, mode='w', newline='', encoding='utf-8') as file:
    writer = csv.DictWriter(file, fieldnames=csv_header)

    # 写入头部
    writer.writeheader()

    # 写入数据行
    for row in csv_data:
        writer.writerow(row)

print(f'CSV file has been saved as {csv_filename} in the current directory.')