Dataframe应用方法返回多个元素（系列）

Question

import pandas as pd

假设我有一个这样的dataframe：

df = pd.DataFrame({"a":range(4),"b":range(1,5)})

看起来是这样的：

   a  b
0  0  1
1  1  2
2  2  3
3  3  4

和一个X乘以Y的函数：

def XtimesY(x,y):
    return x*y

如果我想添加一个新的熊猫系列到df我可以做：

df["c"] =df.apply( lambda x:XtimesY(x["a"],2), axis =1)

它起作用了！

现在我想添加多个系列：

我有这个功能：

def divideAndMultiply(x,y):
    return x/y, x*y

像这样的东西？：

df["e"], df["f"] = df.apply( lambda x: divideAndMultiply(x["a"],2) , axis =1)

它不工作！

我希望'e'列接收除法，'f'列接收乘法！

注意：这不是我正在使用的代码，但我期待相同的行为。

Answer 1

df["e"], df["f"] = zip(*df.apply( lambda x: divideAndMultiply(x["a"],2) , axis =1))

我们应该做到这一点。

（我展示了这个例子，以便您可以看到如何使用多个列作为输入来创建多个新列）

Answer 2

使现代化

更新版本0.23-使用result\u type='broadcast'有关更多详细信息，请参阅文档

重新定义函数，如下所示：

def divideAndMultiply(x,y):
    return [x/y, x*y]

然后这样做：

df[['e','f']] = df.apply(lambda x: divideAndMultiply(x["a"], 2), axis=1, result_type='broadcast')

您将获得期望的结果：

In [118]: df
Out[118]:
   a  b  e  f
0  0  1  0  0
1  1  2  0  2
2  2  3  1  4
3  3  4  1  6

Answer 3

快到了。使用zip*解压函数。试试这个：

def divideAndMultiply(x,y):
    return x/y, x*y

df["e"], df["f"] = zip(*df.a.apply(lambda val: divideAndMultiply(val,2)))