pandas apply 元组分列

将自定义函数返回的元组进行分列，添加到DataFrame的新列中

import pandas as pd

df = pd.DataFrame([range(5), range(5, 10)]).T
df.columns = ['a', 'b']


def add(x, y):
    a = x - y
    b = x + y
    c = x * y
    return a, b, c


df[['计算1', '计算2', '计算3']] = (df.apply(lambda x: add(x['a'], x['b']), axis=1)).apply(pd.Series)
print(df)

'''
   a  b  计算1  计算2  计算3
0  0  5   -5    5    0
1  1  6   -5    7    6
2  2  7   -5    9   14
3  3  8   -5   11   24
4  4  9   -5   13   36
'''