150

我有我认为应该是一个我似乎无法解决的非常简单的任务。

如何将 Python 字典写入 csv 文件？我想要的只是将字典键写入文件的第一行，将键值写入第二行。

我最接近的是以下（我从别人的帖子中得到的）：

f = open('mycsvfile.csv','wb')
w = csv.DictWriter(f,my_dict.keys())
w.writerows(my_dict)
f.close()

问题是，上面的代码似乎只是将键写入第一行，仅此而已。我没有将值写入第二行。

有任何想法吗？

4

2 回答 2

240

您正在使用DictWriter.writerows()which 需要字典列表，而不是字典。你DictWriter.writerow()想写一行。

DictWriter.writeheader()如果您想要 csv 文件的标题，您也将需要使用。

您可能还想查看打开文件的with声明。它不仅更具pythonic和可读性，而且可以为您处理关闭，即使发生异常也是如此。

进行这些更改的示例：

import csv

my_dict = {"test": 1, "testing": 2}

with open('mycsvfile.csv', 'w') as f:  # You will need 'wb' mode in Python 2.x
    w = csv.DictWriter(f, my_dict.keys())
    w.writeheader()
    w.writerow(my_dict)

产生：

test,testing
1,2
于 2012-04-29T15:15:48.150 回答
112

您的代码非常接近工作。

尝试使用常规csv.writer而不是DictWriter。后者主要用于编写字典列表。

这是一些将每个键/值对写入单独行的代码：

import csv

somedict = dict(raymond='red', rachel='blue', matthew='green')
with open('mycsvfile.csv','wb') as f:
    w = csv.writer(f)
    w.writerows(somedict.items())

相反，如果您想要一行中的所有键和下一行中的所有值，那也很容易：

with open('mycsvfile.csv','wb') as f:
    w = csv.writer(f)
    w.writerow(somedict.keys())
    w.writerow(somedict.values())

专业提示：在开发这样的代码时，将编写器设置为，w = csv.writer(sys.stderr)以便您可以更轻松地查看正在生成的内容。当逻辑完善后，切换回w = csv.writer(f).

于 2012-04-29T15:48:58.337 回答