这里需要处理一批数据,希望将本地的数据批量的用 python写入到CSV文件中。这里,我使用 Python 内置 csv 模块。
1、写入列表数据到 CSV
import csv
# 示例数据(二维列表)
data = [
["姓名", "年龄", "城市"],
["张三", 25, "北京"],
["李四", 30, "上海"],
["王五", 22, "广州"]
]
# 写入 CSV 文件
with open("output.csv", mode="w", encoding="utf-8", newline="") as file:
writer = csv.writer(file)
writer.writerows(data) # 写入多行
print("数据已写入 output.csv")
说明:
mode=”w”:写入模式(覆盖原有文件)。
encoding=”utf-8″:避免中文乱码。
newline=””:防止写入多余的空行(Windows 系统必需)。
writer.writerows(data):一次性写入多行数据。
2、写入字典数据到 CSV
如果数据是字典形式,可以使用 csv.DictWriter:
import csv
# 示例数据(字典列表)
data = [
{"姓名": "张三", "年龄": 25, "城市": "北京"},
{"姓名": "李四", "年龄": 30, "城市": "上海"},
{"姓名": "王五", "年龄": 22, "城市": "广州"}
]
# 写入 CSV 文件
with open("output_dict.csv", mode="w", encoding="utf-8", newline="") as file:
fieldnames = ["姓名", "年龄", "城市"] # 表头
writer = csv.DictWriter(file, fieldnames=fieldnames)
writer.writeheader() # 写入表头
writer.writerows(data) # 写入多行数据
print("字典数据已写入 output_dict.csv")
说明:
fieldnames:指定 CSV 的列名(表头)。
writer.writeheader():写入表头。
writer.writerows(data):写入多行字典数据。