如何按键排序不均匀字典并创建CSV

Question

我有一个python字典，其中每个KEY可以有一个可变数量的VALUES（排列在列表中）。

例如：

{'607': [36146], '448': [50890, 44513], '626': [44349, 44436]} 

我想要做的就是生成的这种信息的CSV，象这样的格式：

448 , 607 , 626 
50890,36146,44349 
44513,  ,44436 

目前我的代码可以生成CSV如这是唯一的问题，即CSV的列未按照KEY的升序数字排序。到目前为止我的代码是下面：

csv_file = 'file.csv' 
with open(csv_file, 'wb') as fd: 
    writer = csv.writer(fd, delimiter = ',') 

    # Format headers for aesthetics 
    csv_headers = [' {} '.format(elem) for elem in dictionary.keys()] 

    writer.writerow(headers) 

    # Format data to create convenient csv format 
    csv_data = itertools.izip_longest(*dictionary.values(), fillvalue = '  ') 
    writer.writerows(csv_data) 

正如你看到的我是从价值观分裂密钥，并将它们分开来写，但如果我想通过键的列进行排序我想这可能不是最好的办法去做这件事。因此，我希望有人能指出我正确的（也是最pythonic）方向。

Answer 1

你有两个选择：

• 排序键，然后以相同的顺序提取值，而不是依靠dictionary.values()
• 使用csv.DictWriter() object并产生每行的字典。

选项1是这样的：

csv_file = 'file.csv' 
with open(csv_file, 'wb') as fd: 
    writer = csv.writer(fd, delimiter=',') 

    keys = sorted(dictionary) 
    # Format headers for aesthetics 
    headers = [' {} '.format(key) for key in keys] 
    writer.writerow(headers) 

    # Format data to create convenient csv format 
    csv_data = itertools.izip_longest(*(dictionary[key] for key in keys), 
             fillvalue='  ') 
    writer.writerows(csv_data) 

使用DictWriter会是什么样子：

csv_file = 'file.csv' 
with open(csv_file, 'wb') as fd: 
    writer = csv.DictWriter(
     fd, sorted(dictionary), delimiter=',') 
    # write formatted headers 
    writer.writerow({k: ' {} '.format(k) for k in dicitonary}) 

    csv_data = itertools.izip_longest(*dictionary.values(), fillvalue='  ') 
    writer.writerows(dict(zip(dictionary, row)) for row in csv_data) 

Answer 2

我去整理并用key一个换位元组结束了与的一个iterable名单，然后从那里去：

import csv 
from itertools import izip_longest 

d = {'607': [36146], '448': [50890, 44513], '626': [44349, 44436]} 

with open('output.csv', 'wb') as fout: 
    csvout = csv.writer(fout) 
    header, rows = zip(*sorted((k, iter(v)) for k, v in d.iteritems())) 
    csvout.writerow(format(el, '^5') for el in header) 
    csvout.writerows(izip_longest(*rows, fillvalue='  '))