Skip to main content
 首页 » 编程设计

python之将字典打印到 csv

2025年05月04日16Terrylee

我有一个看起来像这样的字典:

defaultdict(<type 'int'>, {'201304': defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}), '201305': defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}), '201306': defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}), '201301': defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}), '201302': defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}), '201303': defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}), '201212': defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2})}) 

我想要一个看起来像这样的 cvs:

month,District - 1,District -2, District -3...... for however many districts there are  
201304,20,9,5,6,.....  

我现在拥有的是:

with open('output.csv','wb') as output_file: 
    w= csv.writer(output_file) 
    w.writerows(months.items()) 

months 是我上面描述的字典的名称。不幸的是它输出:

201304  defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}) 
201305  defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}) 
201306  defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}) 
201301  defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}) 
201302  defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}) 
201303  defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}) 
201212  defaultdict(<type 'int'>, {'District - 1': 20, 'District - 12': 9, 'District - 6': 5, 'District - 5': 1, 'District - 4': 1, 'District - 3': 1, 'District - 2': 13, 'District - 15': 1, 'District - 14': 2}) 

我需要的是每个月 1 行:

month,District - 1,District -2, District - 3, .... (as many as show up) 
201304,22,34,2,3,4,2,3,14 
201305,34,22,1,3,5 
201306,23,4,42,4,2,2,24,2,5,6,6,7,1 

关于如何修改我的代码以执行此操作的任何建议?

请您参考如下方法:

你可以使用csv.DictWriter,但是因为它是一个二维字典,所以需要添加一个月份列:

#!python3 
from collections import defaultdict 
import csv 
 
# populate the defaultdict "months" 
# <deleted to keep short> 
 
# Build a set of unique districts 
s = set() 
for k,v in months.items(): 
    s.update(v.keys()) 
 
# open per requirements of csv.writer 
with open('out.csv','w',newline='') as f: 
    # wrap in a DictWriter and specify column names 
    w = csv.DictWriter(f,fieldnames=['month']+list(sorted(s))) 
    w.writeheader() 
    for k,v in months.items(): 
        # copy the dict of districts, add a month key and write the row 
        temp = dict(v) 
        temp['month'] = k 
        w.writerow(temp) 

输出:

month,District - 1,District - 12,District - 14,District - 15,District - 2,District - 3,District - 4,District - 5,District - 6 
201301,20,9,2,1,13,1,1,1,5 
201302,20,9,2,1,13,1,1,1,5 
201303,20,9,2,1,13,1,1,1,5 
201304,20,9,2,1,13,1,1,1,5 
201305,20,9,2,1,13,1,1,1,5 
201306,20,9,2,1,13,1,1,1,5 
201212,20,9,2,1,13,1,1,1,5 

注意:如果使用 Python 2,请改用以下 open:

with open('out.csv','wb') as f: