我认为应该是一项非常简单的任务,但我似乎无法解决。
如何将 Python 字典写入 csv 文件?我想要的只是将字典键写入文件的顶行,将键值写入第二行。
我最接近的是以下内容(我从别人的帖子中得到的):
f = open('mycsvfile.csv','wb')
w = csv.DictWriter(f,my_dict.keys())
w.writerows(my_dict)
f.close()
问题是,上面的代码似乎只是将键写入第一行,仅此而已。我没有将值写入第二行。
有什么想法吗?
您正在使用
DictWriter.writerows()
,它需要一个字典列表,而不是一个字典。您希望 DictWriter.writerow()
写入一行。
如果您想要 csv 文件的标题,您还需要使用
DictWriter.writeheader()
。
with
声明以获取打开文件。它不仅更具Python风格和可读性,而且可以为您处理关闭,即使发生异常也是如此。
进行这些更改的示例:
import csv
my_dict = {"test": 1, "testing": 2}
with open("mycsvfile.csv", "w", newline="") as f:
w = csv.DictWriter(f, my_dict.keys())
w.writeheader()
w.writerow(my_dict)
产生:
test,testing
1,2
如评论中所述,我们使用
newline=""
作为 CSV 编写器进行自己的换行管理,因此我们禁用 open()
中的内置管理,如文档中所述。
您的代码非常接近工作。
尝试使用常规的 csv.writer 而不是 DictWriter。后者主要用于编写字典列表。
这里有一些代码,将每个键/值对写入单独的行:
import csv
somedict = dict(raymond='red', rachel='blue', matthew='green')
with open('mycsvfile.csv','w') as f:
w = csv.writer(f)
w.writerows(somedict.items())
如果您希望所有键都在一行上,所有值都在下一行上,这也很简单:
with open('mycsvfile.csv','w') as f:
w = csv.writer(f)
w.writerow(somedict.keys())
w.writerow(somedict.values())
专业提示:开发此类代码时,请将 writer 设置为
w = csv.writer(sys.stderr)
,以便您可以更轻松地查看正在生成的内容。当逻辑完善后,切换回w = csv.writer(f)
。