Python数据透视表：基于现有结果计算添加百分比列的方法

原始数据表 #

after   before
a   200     630
b   750     852
c   457     76738
d   3205    32235
e   12408   188561
f   1192    497130

需求说明 #

需为数据表添加percent列，计算规则为：percent = after/(after+before)

预期结果 #

after   before  percent
a   200     630     0.240963855
b   750     852     0.468164794
c   457     76738   0.005920073
d   3205    32235   0.090434537
e   12408   188561  0.061740866
f   1192    497130  0.002392028

函数实现补充 #

基于给定框架，补充后的完整函数如下：

import numpy as np

def get_percent(x):
    " for each value in row, return a percentage by after/(after+before) corresponds to "
    percent = x['after'] / (x['after'] + x['before'])
    return np.array([percent])

说明 #

函数接收DataFrame的行对象x，提取该行的after和before数值，按公式计算得到percent值后，返回包含该值的numpy数组，符合原函数的返回要求。若要为数据表添加该列，可调用df['percent'] = df.apply(get_percent, axis=1)（需确保df为目标数据表对象）。

内容的提问来源于stack exchange，提问作者fatcat