Python：将嵌套字典写入CSV

Question

我正在尝试将嵌套的字典写入.csv文件。这是一个简单的示例：

import csv
import itertools

fields = [ 'org', '2015', '2014', '2013' ]
dw     = { 'orgname1': { '2015' : 2, '2014' : 1, '2013' : 1 },
           'orgname2': { '2015' : 1, '2014' : 2, '2013' : 3 },
           'orgname3': { '2015' : 1, '2014' : 3, '2013' : 1 }
        }

with open("test_output.csv", "wb") as f:
    w = csv.writer( f )
    years = dw.values()[0].keys()
    for key in dw.keys():
        w.writerow([key, [dw[key][year] for year in years]])

这使我得到一个包含两列的表：第一个包含orgname;
第二个包含[2，1，1]（或子词典中的相应值）。我想要一个有四列的表：一列orgname对应的列表元素，然后三列对应的列表元素。

Answer 1

更改：

w.writerow([key, [dw[key][year] for year in years]])

至：

w.writerow([key] + [dw[key][year] for year in years])

否则，您会尝试向[orgname1, [2, 1, 1]]csv编写类似的内容，而您的意思是[orgname1, 2, 1, 1]。

如Padraic所述，您可能希望更改years = dw.values()[0].keys()为years = sorted(dw.values()[0].keys())或years = fields[1:]避免随机行为。