python - 将新列插入到数据框中，该数据框根据超过 1 个变量进行计算

Question

我想根据相应列的计算将新列插入到数据框中，

这对我有用:

df['new column'] = [ foo(x) for x in df['X']]

现在，如果我们想要为 > 1 列进行计算

df['new column'] = [ foo(x, y, z) for x, y, z in df['X', 'Y', 'Z']]

这是我尝试过的，但它不接受语法。我可以迭代，但我想内联计算。

有人可以帮忙吗？

Answer 1

您可以直接访问底层的numpy数组:

df['new column'] = [foo(x, y, z) for x, y, z in df[['X', 'Y', 'Z']].values]

请注意，这可能是您想要的(并且它可能比 iterrows 或 apply 更快)，但您无法访问 numpy 数组中列名的索引。