如何使用csv.write输出到csv文件并添加头文件- Python3/boto3

import csv

# 假设这是你的数据
data = [
    {'name': 'Alice', 'age': 30, 'city': 'New York'},
    {'name': 'Bob', 'age': 25, 'city': 'San Francisco'},
    {'name': 'Charlie', 'age': 35, 'city': 'Chicago'}
]

# 表头字段
fieldnames = ['name', 'age', 'city']

# 写入CSV文件
with open('output.csv', mode='w', newline='') as csv_file:
    writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
    
    # 写入表头
    writer.writeheader()
    
    # 写入数据
    for row in data:
        writer.writerow(row)

import csv
import boto3
from io import StringIO

# 初始化S3客户端
s3_client = boto3.client('s3')

# 假设这是你的数据
data = [
    {'name': 'Alice', 'age': 30, 'city': 'New York'},
    {'name': 'Bob', 'age': 25, 'city': 'San Francisco'},
    {'name': 'Charlie', 'age': 35, 'city': 'Chicago'}
]

# 表头字段
fieldnames = ['name', 'age', 'city']

# 使用StringIO作为内存中的文件对象
output = StringIO()
writer = csv.DictWriter(output, fieldnames=fieldnames)

# 写入表头
writer.writeheader()

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

# 获取内存中文件的值
csv_data = output.getvalue()

# 将数据上传到S3
s3_client.put_object(Bucket='your-bucket-name', Key='output.csv', Body=csv_data)

# 关闭StringIO对象
output.close()