將字典列表寫到csv中，其中字典值是列表

Question

這是我到目前為止的代碼；

list_of_dict = [{'A': [1, 2, 3, 4, 5], 'B': [10, 11]}]
with open('result.csv','wb') as out:
    f1 = csv.DictWriter(out,list_of_dict[0].keys())
    f1.writeheader()
    f1.writerows(list_of_dict)

我得到的輸出是；

     A            B
[1,2,3,4,5]    [10,11]

我怎么得到的輸出為;

A   B
1   10
2   11
3
4
5

Answer 1

您必須將您的一本字典轉置成字典列表，輸出中的每一行都對應一個字典：

import csv
from itertools import izip_longest

with open('result.csv','wb') as out:
    keys = list_of_dict[0].keys()
    f1 = csv.DictWriter(out, keys)
    f1.writeheader()
    f1.writerows(dict(zip(keys, row)) for row in izip_longest(*list_of_dict[0].values()))

請注意，這些鍵將以字典（任意）順序列出。

上面的生成器表達式按行生成所需的字典：

>>> from itertools import izip_longest
>>> list_of_dict = [{'A': [1, 2, 3, 4, 5], 'B': [10, 11]}]
>>> keys = list_of_dict[0].keys()
>>> [dict(zip(keys, row)) for row in izip_longest(*list_of_dict[0].values())]
[{'A': 1, 'B': 10}, {'A': 2, 'B': 11}, {'A': 3, 'B': None}, {'A': 4, 'B': None}, {'A': 5, 'B': None}]