Python使用csv.writer写入值为列表的字典时无法生成目标CSV布局
问题说明
需要将值为列表类型的字典数据写入符合标准格式的CSV文件,确保文件可被MySQL正常识别。
使用的测试字典结构如下:
{ 'dillbill.grp': ['dillbiii', 'dilljt', 'dilltrst', 'dillwren', 'dillwind', 'dillgeor', 'dillgind', 'dillthom', 'dillells'], 'zgrptest.grp': ['dillbiii', 'dilljt', 'dilltrst', 'dillwren', 'dillwind', 'dillgeor', 'dillgind', 'dillthom', 'dillells', 'achfdn'] }
预期输出布局为两列结构:第一列为字典的键,第二列为该键对应列表中的单个元素,每一组键+列表元素单独占一行。
之前使用print拼接写入的代码虽然布局符合要求,但没有遵循CSV标准转义规则,且逗号前后会自动添加多余空格,特殊字符场景下会导致MySQL解析失败。原有print实现代码如下:
with open('groups_old.csv','w', encoding='UTF8', newline='') as datacsv: for key, values in dicto.items(): for v in values: print(key,',',v, file=datacsv)
原有错误的csv.writer实现存在三个问题:遍历字典items时未正确解包键值对、混淆writerow()/writerows()的用法、未指定文件编码和newline参数导致格式错乱。错误代码如下:
with open('groups.csv','w') as datacsv: writer=csv.writer(datacsv) for keys in dicto.items(): writer.writerows(key) writer.writerows(values)
正确实现代码
按照csv模块标准用法修正后的代码如下,生成的文件完全符合CSV格式规范,可直接被MySQL识别导入:
import csv # 目标字典数据 dicto = { 'dillbill.grp': ['dillbiii', 'dilljt', 'dilltrst', 'dillwren', 'dillwind', 'dillgeor', 'dillgind', 'dillthom', 'dillells'], 'zgrptest.grp': ['dillbiii', 'dilljt', 'dilltrst', 'dillwren', 'dillwind', 'dillgeor', 'dillgind', 'dillthom', 'dillells', 'achfdn'] } with open('groups.csv', 'w', encoding='UTF8', newline='') as datacsv: writer = csv.writer(datacsv) for key, value_list in dicto.items(): for single_val in value_list: # 逐行写入两列内容,csv模块自动处理分隔和转义 writer.writerow([key, single_val])
关键说明
- 打开文件时必须指定
newline='',为Python csv模块官方要求的写法,可避免跨平台使用时出现多余空行 - 写入单行数据使用
writerow()方法,传入的列表每个元素对应CSV的一列;writerows()用于批量写入多行,要求传入的参数是嵌套的可迭代对象,直接传入字符串会将每个字符拆分为单独列 with上下文管理器会自动完成文件关闭操作,不需要手动调用close()- csv模块会自动处理包含特殊字符(逗号、引号、换行符等)的字段,生成的文件完全符合RFC4180标准,不会出现MySQL识别异常的问题
内容的提问来源于stack exchange,提问作者LRSpartan
相关产品推荐
相关产品推荐

