我有以下数据框:
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个数字相加,然后将它们除以所有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
我该怎么办?
答案 0 :(得分:2)
使用列表理解:
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]