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

這看起來像是DictWriter的工作：

import csv
import itertools
import sys

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 }
        }

w = csv.DictWriter( sys.stdout, fields )
for key,val in sorted(dw.items()):
    row = {'org': key}
    row.update(val)
    w.writerow(row)

Answer 2

使用DictWriter和標頭的替代實現

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.DictWriter(f, fields)
    w.writeheader()
    for k in dw:
        w.writerow({field: dw[k].get(field) or k for field in fields})

輸出：

org,2015,2014,2013
orgname1,2,1,1
orgname3,1,3,1
orgname2,1,2,3

Answer 3

改變：

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:]到避免隨機行為。

Answer 4

使用 DictWriter 不需要提前對字段進行排序，因為w.writerow()將確保正確的順序。 但對項目本身進行排序確實有意義。

因此，將上述所有建議放在一起並從中挑選最好的，我會想出以下代碼：

import csv
import itertools

def mergedict(a,b):
    a.update(b)
    return a

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.DictWriter( f, fields )
    w.writeheader()
    for k,d in sorted(dw.items()):
        w.writerow(mergedict({'org': k},d))

我添加了一個微小的mergedict()函數，使它成為一個進一步向下的班輪。

Answer 5

我認為這可能是一種更簡單的方法：

import csv

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", "w") as csv_file:
  csvwriter = csv.writer(csv_file)
  csvwriter.writerow(['org', '2015', '2014', '2013'])

  for org in dw:
     csvwriter.writerow(org, dw[org]['2015'], dw[org]['2014'], dw[org]['2013'])