熊猫数据框-在列中的列表中查找总和

Question

I have a list like

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]} ]

I need to sum every digit in list column:

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]`

And in final:

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

How can I do this?

Answer 1

You can use this to sum a list of dict of list:

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

You can easily change the entries into numpy.array s, then sum:

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]