对数据框熊猫的列中存在的列表执行计算

Question

我有以下数据框：

    Position        A           B   
0   29644164       71.0    [31, 38, 1, 1]   
1   45861974       45.0    [17, 26, 1, 1]   
2   58142396       69.0    [37, 31, 0, 1]   
3   41223046       75.0    [21, 53, 0, 1] 

我想对B列进行计算。我想将2个数字加到最后2个位置，然后将它们除以所有4个数字的和。 然后，将结果添加到新列“ calc”中

因此，我期望的数据帧将是：

     Position        A           B          calc
0   29644164       71.0    [3, 5, 1, 1]     0.2
1   45861974       45.0    [2, 2, 1, 1]     0.333
2   58142396       69.0    [3, 7, 0, 1]     0.1
3   41223046       75.0    [3, 2, 0, 1]     0.2

我该怎么做？

Answer 1

使用清单理解：

df['calc'] = [sum(x[-2:]) / sum(x) for x in df.B]
print (df)
   Position     A               B      calc
0  29644164  71.0  [31, 38, 1, 1]  0.028169
1  45861974  45.0  [17, 26, 1, 1]  0.044444
2  58142396  69.0  [37, 31, 0, 1]  0.014493
3  41223046  75.0  [21, 53, 0, 1]  0.013333

详细资料 ：

print ([sum(x[-2:]) for x in df.B])
[2, 2, 1, 1]

print ([sum(x) for x in df.B])
[71, 45, 69, 75]