功能
将get_answer()函数得到的dict存入到csv表格
需注意点
- 写入csv文件需要加参数 newline=‘’,可保证写入csv文件中没有空行
- 写入csv文件需加参数 encoding=‘utf-8’,转dict为str调用json.dumps()函数需加参数 ensure_ascii=False,可保证写入csv的汉字正常显示
代码
with open(r'output.csv', 'w', newline='', encoding='utf-8') as f:
for query in tqdm(column):
answer_dict = get_answer(query)
answer = json.dumps(answer_dict, ensure_ascii=False)
writer = csv.writer(f)
writer.writerow([answer])