- 科研需要,用大语言模型生成了一系列文本数据集,现需要将其整理为csv文件,python代码实现如下(简要版)
import csv
from datetime import datetime
data = [
{"日期": "2024-05-13", "时间戳": "08:00", "情绪": "平静", "地点": "家里", "记忆流内容": "早上好,今天天气很好,准备出门散步。"},
{"日期": "2024-05-13", "时间戳": "12:30", "情绪": "焦虑", "地点": "公司", "记忆流内容": "忙碌的一天开始了,先处理一些邮件和会议。"},
{"日期": "2024-05-13", "时间戳": "13:25", "情绪": "平静", "地点": "健身房", "记忆流内容": "下班后去健身房锻炼身体,放松一下。"},
{"日期": "2024-05-13", "时间戳": "14:40", "情绪": "开心", "地点": "餐厅", "记忆流内容": "和朋友一起吃晚饭,聊聊近期的生活。"}
]
with open("data.csv", "w", newline="", encoding="utf-8") as csvfile:
fieldnames = ["日期", "时间戳", "情绪", "地点", "记忆流内容"]
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
for row in data:
writer.writerow(row)
print("CSV文件已生成。")