编写字典列表,每个键有多个值作为新行

时间:2017-03-17 18:51:18

标签: python python-3.x csv dictionary

作为前言,会有很多重复的问题,但没有一个解决方案对我有帮助。我使用通常的代码编写一个字典列表,使用csv包:

    import csv
    to_csv = list_dic
    keys = to_csv[0].keys()
    with open('output.csv', 'w') as output_file:
        dict_writer = csv.DictWriter(output_file, keys)
        dict_writer.writeheader()
        dict_writer.writerows(to_csv)

这个代码在单个键值的字典列表上使用时很好(除了输出csv每隔一行有一个空行之外),但现在我有一个包含多个值的字典列表每个键,非常类似于下面的示例:

    list_dic = [{'a':[x,y],'b':[1,2],'c':[1,2]},
                {'a':[x,y],'b':[1,2],'c':[1,2]},
                {'a':[x,y],'b':[1,2],'c':[1,2]},
                {'a':[x,y],'b':[1,2],'c':[1,2]}]

基本上我的csv输出目前看起来像这样:

     'a'     'b'     'c'
    [x,y]   [1,2]   [1,2]
    [x,y]   [1,2]   [1,2]
    [x,y]   [1,2]   [1,2]
    [x,y]   [1,2]   [1,2]

我正试着到这里来:

      'a'    'b'    'c'
       x      1      1
       y      2      2
       x      1      1
       y      2      2
       x      1      1
       y      2      2
       x      1      1
       y      2      2          

我倾注了非常相似的问题,但似乎无法将任何解决方案应用于我的特定情况。我尝试使用某种for语句来写多行,但是失败了。

感谢您的帮助,如果您知道任何与此情况重复的问题,请指出正确的方向。

编辑:示例每个键有2个值,但实际上值的数量是未知的,只是> 1

4 个答案:

答案 0 :(得分:3)

如果您足够压缩和解压缩数据,您将获得所需的格式:

list_dic = [{'a':['x','y'],'b':[1,2],'c':[1,2]},
            {'a':['x','y'],'b':[1,2],'c':[1,2]},
            {'a':['x','y'],'b':[1,2],'c':[1,2]},
            {'a':['x','y'],'b':[1,2],'c':[1,2]}]
import csv
to_csv = list_dic
keys = to_csv[0].keys()
with open('output.csv', 'w') as output_file:
    dict_writer = csv.DictWriter(output_file, keys)
    dict_writer.writeheader()
    for dic in list_dic:
        keys, values = zip(*dic.items())
        for values in zip(*values):
            dict_writer.writerow(dict(zip(keys, values)))

答案 1 :(得分:1)

拆分字典,然后通过csv运行它们:

def split_dict(dct):
    result = []
    result.append({k: v[0] for k, v in dct.items()})
    result.append({k: v[1] for k, v in dct.items()})
    return result

def list_dict_split(lst):
    result = []
    for dct in lst:
        result.extend(split_dict(dct))
    return result

现在只需将list_dict_split(list_dic)放在list_dic:

的任何地方

中间名单是:

[{'a': 'x', 'c': 1, 'b': 1}, 
 {'a': 'y', 'c': 2, 'b': 2}, 
 {'a': 'x', 'c': 1, 'b': 1}, 
 {'a': 'y', 'c': 2, 'b': 2}, 
 {'a': 'x', 'c': 1, 'b': 1}, 
 {'a': 'y', 'c': 2, 'b': 2}, 
 {'a': 'x', 'c': 1, 'b': 1}, 
 {'a': 'y', 'c': 2, 'b': 2}]

答案 2 :(得分:1)

这是另一种选择,只需要一个zipcolumns转换为rows

x = 'x'
y = 'y'

list_dic = [{'a': [x, y], 'b':[1, 2], 'c':[1, 2]},
            {'a': [x, y], 'b':[3, 4], 'c':[1, 2]},
            {'a': [x, y], 'b':[1, 2], 'c':[1, 2]},
            {'a': [x, y], 'b':[3, 4], 'c':[1, 2]}]

keys = ['a', 'b', 'c']
sep = "\t"

print(sep.join(keys))
for dic in list_dic:
    columns = [dic[key] for key in keys]
    for row in zip(*columns):
        print(sep.join(str(cell) for cell in row))

输出:

a   b   c
x   1   1
y   2   2
x   3   1
y   4   2
x   1   1
y   2   2
x   3   1
y   4   2

答案 3 :(得分:0)

您可以将当前的词典列表转换为爆炸词典列表,然后将其写入csv:

>>> import itertools as it
>>> [dict(n) for d in list_dic for n in zip(*(zip(it.repeat(k), v) for k, v in d.items()))]
[{'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 2, 'c': 2},
 {'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 2, 'c': 2},
 {'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 2, 'c': 2},
 {'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 2, 'c': 2}]

这适用于任意数量的值:

>>> list_dic = [{'a':['x','y','z'],'b':[1,2,3],'c':[1,2,3]},
...             {'a':['x','y','a'],'b':[1,3,2],'c':[1,2,1]},
...             {'a':['x','y','b'],'b':[1,4,3],'c':[1,2,4]},
...             {'a':['x','y','a'],'b':[1,5,2],'c':[1,2,9]}]
>>> [dict(n) for d in list_dic for n in zip(*(zip(it.repeat(k), v) for k, v in d.items()))]
[{'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 2, 'c': 2},
 {'a': 'z', 'b': 3, 'c': 3},
 {'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 3, 'c': 2},
 {'a': 'a', 'b': 2, 'c': 1},
 {'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 4, 'c': 2},
 {'a': 'b', 'b': 3, 'c': 4},
 {'a': 'x', 'b': 1, 'c': 1},
 {'a': 'y', 'b': 5, 'c': 2},
 {'a': 'a', 'b': 2, 'c': 9}]