我今天在工作中遇到了这个问题,我想知道是否有一种有效的方法可以做到这一点。

基本上我有一个看起来像这样的dataframe

df = pd.DataFrame([1,2,3], columns = ['a'])

我还有一个返回元组的function。 (请注意,这只是一个最小的例子,我的问题是不同的)

compute = lambda x: (2*x, 3*x)

我需要找到一种方法来做一些理想情况下的事情:

(df['b'], df['c']) = df['a'].apply(compute)

不幸的是,这种语法不起作用,我不能用另一种方式来解决这个问题。

唯一类似的问题是这个 ,但解决方案似乎真的“hacky”,我相信有更好的方法来做到这一点。

谢谢!!

分析解答

IIUC,您可以尝试:

compute = lambda x: (2*x, 3*x)
df[['b','c']]=pd.DataFrame(df.a.apply(compute).tolist()) #thanks harvpan
#df[['b','c']]=pd.DataFrame(df.a.apply(compute).values.tolist())
print(df)

   a  b  c
0  1  2  3
1  2  4  6
2  3  6  9