Pandas：根据列中的值向 DataFrame 添加新列

Question

给定这样的 DataFrame：

>>> df
    0   1   2
0   2   3   5
1   3   4   7

还有一个返回多个结果的函数，像这样：

def sumprod(x, y, z):
    return x+y+z, x*y*z

我想添加新列，所以结果是：

>>> df
    0   1   2  sum  prod
0   2   3   5   10    30
1   3   4   7   14    84

我已经成功使用返回一个结果的函数：

df["sum"] = p.apply(sum, axis=1)

但如果它返回多个结果，则不会。

Answer 1

一种方法是通过解包数组的转置将 DataFrame 的列传递给函数：

>>> df['sum'], df['prod'] = sumprod(*df.values.T)
>>> df
   0  1  2  sum  prod
0  2  3  5   10    30
1  3  4  7   14    84

sumprod 返回一个列元组，由于 Python 支持多重赋值，您可以将它们分配给新的列标签，如上所述。

你可以写df['sum'], df['prod'] = sumprod(df[0], df[1], df[2]) 来得到同样的结果。如果您需要以特定顺序将列传递给函数，这会更清晰并且更可取。另一方面，如果您有很多列要传递给函数，则会更加冗长。