Question

我有一个像

这样的清单

source = [{'TGA': [0, 1, 0, 0], 'AAC': [0, 0, 0, 1], 'GAA': [0, 0, 1, 0], 
           'GTG': [1, 0, 0, 0]},{'TGA': [0, 1, 0, 0], 'AAC': [0, 0, 0, 1], 
           'GAA': [0, 0, 1, 0], 'GTG': [1, 0, 0, 0]} ]

我需要对列表列中的每个数字求和：

pandas.DataFrame(source)
        AAC           GAA           GTG           TGA
  0  [0, 0, 0, 1]  [0, 0, 1, 0]  [1, 0, 0, 0]  [0, 1, 0, 0]
  1  [0, 0, 0, 1]  [0, 0, 1, 0]  [1, 0, 0, 0]  [0, 1, 0, 0]`

最后：

         AAC           GAA           GTG           TGA
    sum  [0, 0, 0, 2 ] [0, 0, 2, 0] [2, 0, 0, 0]  [0, 2, 0, 0]

我该怎么做？

Answer 1

您可以使用它来汇总列表的字典列表：

source = [{'TGA': [0, 1, 0, 0], 'AAC': [0, 0, 0, 1], 'GAA': [0, 0, 1, 0],
           'GTG': [1, 0, 0, 0]},{'TGA': [0, 1, 0, 0], 'AAC': [0, 0, 0, 1],
           'GAA': [0, 0, 1, 0], 'GTG': [1, 0, 0, 0]} ]

res = {}
for d in source:
    for key,value in d.items():
            if key not in res:
                    res[key] = value
            else:
                    res[key] = [x+y for x,y in zip(res[key],value) ]

print res

Answer 2

您可以轻松地将条目更改为numpy.array s，然后加上：

import numpy as np

>> df.applymap(np.array).sum()
AAC    [0, 0, 0, 2]
GAA    [0, 0, 2, 0]
GTG    [2, 0, 0, 0]
TGA    [0, 2, 0, 0]

Pandas Dataframe - 在列中的列表中查找总和

2 个答案: