將嵌套字典寫入csv python

Question

我正在嘗試將嵌套字典寫入python。 我有如下字典：

{'09-04-2018' : {1: 11, 2: 5, 3: 1, 4: 1, 5: 0} , '10-04-2018' : {1: 5, 2: 1, 3: 1, 4: 1, 5: 0}}

我想寫這樣的東西：

count,09-04-2018,10-04-2018
1,11,5
2,5,1
3,1,1
4,1,1
5,0,0

Answer 1

以下產生請求的輸出：

data = {'09-04-2018' : {1: 11, 2: 5, 3: 1, 4: 1, 5: 0} , '10-04-2018' : {1: 5, 2: 1, 3: 1, 4: 1, 5: 0}}

rows = []
keys = sorted(data)
header = ['count'] + keys
counts = sorted(set(k for v in data.values() for k in v))
for count in counts:
    l = [count]
    for key in keys:    
        l.append(data[key].get(count))
    rows.append(l)

print header
print rows

import csv
with open('output.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(header)
    writer.writerows(rows)

這樣在寫行之前先建立行，可以直接寫行，而不是將它們附加到列表中並寫列表的內容。

產生以下輸出：

count,09-04-2018,10-04-2018
1,11,5
2,5,1
3,1,1
4,1,1
5,0,0

Answer 2

如果您願意使用第三方庫，則可以使用pandas ：

import pandas as pd

d = {'09-04-2018' : {1: 11, 2: 5, 3: 1, 4: 1, 5: 0},
     '10-04-2018' : {1: 5, 2: 1, 3: 1, 4: 1, 5: 0}}

# create dataframe from dictionary
df = pd.DataFrame.from_dict(d).reset_index().rename(columns={'index': 'count'})

# write dataframe to csv file
df.to_csv('file.csv', index=False)

print(df)

#    count  09-04-2018  10-04-2018
# 0      1          11           5
# 1      2           5           1
# 2      3           1           1
# 3      4           1           1
# 4      5           0           0

Answer 3

您可以使用zip縮短代碼：

import csv
d = {'09-04-2018' : {1: 11, 2: 5, 3: 1, 4: 1, 5: 0} , '10-04-2018' : {1: 5, 2: 1, 3: 1, 4: 1, 5: 0}}
with open('filename.csv', 'w') as f:
  write = csv.writer(f)
  full_rows = [i for h in [zip(*b.items()) for _, b in sorted(d.items(), key=lambda x:map(int, x[0].split('-')))] for i in h]
  write.writerows([['counts']+[a for a, _ in sorted(d.items(), key=lambda x:map(int, x[0].split('-')))]]+list(zip(*[full_rows[0]]+full_rows[1::2])))

輸出：

counts,09-04-2018,10-04-2018
1,11,5
2,5,1
3,1,1
4,1,1
5,0,0