我有一个由两列组成的数据框架,每列都包含列表元素。我想对两个列表执行行元素加法。我下面的解决方案就是从这个答案中得到启发的
Element-wise addition of 2 lists?
我的代码:
df = pd.DataFrame({'A':[[1,2,3],[10,20]],'B':[[4,5,6],[40,50]]})
A B
0 [1, 2, 3] [4, 5, 6]
1 [10, 20] [40, 50]
df['C'] = df[['A','B']].apply(lambda x: list(map(add,x[0],x[1])))
df['C'] =
A B
0 11 44
1 22 55
预期答案:
df['C'] =
C
0 [5,7,9]
1 [50,70]
