Добавьте 2 новых столбца в существующий фрейм данных, используя apply

Я хочу использовать функцию apply, которая: - принимает 2 столбца в качестве входных данных - выводит два новых столбца в зависимости от функции.

Пример с этой функцией add_multiply.

#function with 2 column inputs and 2 outputs
def add_multiply (a,b):
  return (a+b, a*b )

#example dataframe
df = pd.DataFrame({'col1': [1, 2], 'col2': [3, 4]})

#this does not work
df[['add', 'multiply']] = df.apply(lambda x: add_multiply(x['col1'], x['col2']), axis=1)

идеальный результат:

col1  col2  add  multiply
1     3     4    3
2     4     6    8

Ответы

Ответ 1

Вы можете добавить result_type='expand' в apply:

‘expand: list-like results will be turned into columns.

df[['add', 'multiply']]=df.apply(lambda x: add_multiply(x['col1'], x['col2']),axis=1,
                             result_type='expand')

Или вызовите конструктор данных:

df[['add', 'multiply']]=pd.DataFrame(df.apply(lambda x: add_multiply(x['col1'], 
                                    x['col2']), axis=1).tolist())

   col1  col2  add  multiply
0     1     3    4         3
1     2     4    6         8

Ответ 2

Ответ anky_91 выделяет полезную опцию в apply.

Однако для этого конкретного случая apply даже не требуется,

df['add'], df['multiply'] = add_multiply(df['col1'],df['col2'])