Python：从{(元组)：值}字典的'列表'写入csv文件

Question

大家好。

我有一个列表，里面的元素是这样的字典：[(元组): 值]，具体如下：

lst = [{('unit1', 'test1'): 11, ('unit1','test2'): 12}, {('unit2','test1'): 13, ('unit2','test2'):14 }]

testnames = ['test1','test2']
unitnames = ['unit1','unit2']

我该如何将这些数据写入一个csv文件，输出格式如下呢？

单位名称, 测试1, 测试2

单位1, 11, 12

单位2, 13, 14

谢谢。

Answer 1

下面的代码创建了一个字典，这个字典可以被导出成一个csv文件：

from collections import defaultdict
d = defaultdict(list)
for (unit, test), val in lst.items():
    d[unit].append(val)

Answer 2

首先，把结构变得简单明了。

units = collections.defaultdict(lambda: collections.defaultdict(lambda: float('-inf')))
for (u, t), r in lst.iteritems():
  units[u][t] = r
table = [(u, t['test1'], t['test2']) for (u, t) in units.iteritems()]

然后使用 csv 来写出CSV文件。

Answer 3

你把列表（lst）分组的方式有点多余；所有的键都是唯一的，完全可以用一个字典来表示，像这样：

data = {
    ('unit1', 'test1'): 11,
    ('unit1', 'test2'): 12,
    ('unit2', 'test1'): 13, 
    ('unit2', 'test2'): 14
}

这样做之后

import csv

def getUniqueValues(seq):
    "Return sorted list of unique values in sequence"
    values = list(set(seq))
    values.sort()
    return values

def dataArray(data2d, rowIterField=0, rowLabel='', defaultVal=''):
    # get all unique unit and test labels
    rowLabels = getUniqueValues(key[rowIterField] for key in data2d)
    colLabels = getUniqueValues(key[1-rowIterField] for key in data2d)

    # create key-tuple maker
    if rowIterField==0:
        key = lambda row,col: (row, col)
    else:
        key = lambda row,col: (col, row)

    # header row
    yield [rowLabel] + colLabels
    for row in rowLabels:
        # data rows
        yield [row] + [data2d.get(key(row,col), defaultVal) for col in colLabels]

def main():
    with open('output.csv', 'wb') as outf:
        outcsv = csv.writer(outf)
        outcsv.writerows(dataArray(data, 0, 'unitnames'))

if __name__=="__main__":
    main()

输出的格式也很容易调整（单位在横向，测试在纵向），只需要把 dataArray(data, 0, 'unitnames') 改成 dataArray(data, 1, 'testnames') 就可以了。