Предположим, что у меня есть df
, который имеет столбцы из 'ID', 'col_1', 'col_2'
. И я определяю функцию:
f = lambda x, y : my_function_expression
.
Теперь я хочу применить f
to df
два столбца 'col_1', 'col_2'
для элементарного вычисления нового столбца 'col_3'
, несколько как:
df['col_3'] = df[['col_1','col_2']].apply(f)
# Pandas gives : TypeError: ('<lambda>() takes exactly 2 arguments (1 given)'
Как сделать?
** Добавьте пример образца ниже ***
import pandas as pd
df = pd.DataFrame({'ID':['1','2','3'], 'col_1': [0,2,3], 'col_2':[1,4,5]})
mylist = ['a','b','c','d','e','f']
def get_sublist(sta,end):
return mylist[sta:end+1]
#df['col_3'] = df[['col_1','col_2']].apply(get_sublist,axis=1)
# expect above to output df as below
ID col_1 col_2 col_3
0 1 0 1 ['a', 'b']
1 2 2 4 ['c', 'd', 'e']
2 3 3 5 ['d', 'e', 'f']