带有功能的Pandas DataFrame逐行公式

Question

所以我有相当大的DataFrame，并根据其他列根据一些方程式创建了一个新列：

df['F'] = (params.a * params.b * df.A/1000 - param.C * (df.B + df.C - df.D) + param.D * df.E

而且效果很好。 除了要在整个代码中重复此功能之外 ，因此我不想将其易于复制和粘贴，而是将其强制转换为可重用的功能。

所以我将其转换为lambda：

def fun(r):
     return (params.a * params.b * r.A/1000 - param.C * (r.B + r.C - r.D) + param.D * r.E   
df['F'] = r.apply(funy,axis =1)

但这现在慢了5倍（对于1万行，是1.2秒与6秒 ）。

如果我想拥有整洁的功能并且仍然受益于速度，该怎么办？

Answer 1

有什么问题：

def fun():
    return params.a * params.b * df.A/1000 - param.C * (df.B + df.C - df.D) + param.D * df.E

df['F'] = fun()

因此，您将获得可重用的矢量化函数。