python – 从动态字典创建一个矩阵

输入：

data = [
    {'a': 2,'g': 1},{'p': 3,'a': 5,'cat': 4}
    ...
]

输出：

a  p  cat  g
1st  2  0  0    1
2nd  5  3  4    0

这是我的代码.但是,当数据量巨大时,我认为它并不聪明且非常慢.

有什么好方法可以做到这一点吗？

谢谢.

data = [
    {'a': 2,'cat': 4}
]

### Get keyword map ###
key_map = set()
for row in data:
    key_map = key_map.union(set(row.keys()))

key_map = list(key_map)    # ['a','p','g','cat']

### Create matrix ###
result = []
for row in data:
    matrix = [0] * len(key_map)
    for k,v in row.iteritems():
        matrix[key_map.index(k)] = v
    result.append(matrix)

print result        

# [[2,1],[5,3,4,0]]

编辑

通过@wwii建议.使用Pandas看起来不错：

from pandas import DataFrame

result = DataFrame(data,index=range(len(data)))
print result.fillna(0,downcast=int).as_matrix().tolist()
# [[2,1,0],3]]