pandas apply function with parameters from multiple columns code example
Example 1: apply a function to multiple columns in pandas
In [49]: df
Out[49]:
0 1
0 1.000000 0.000000
1 -0.494375 0.570994
2 1.000000 0.000000
3 1.876360 -0.229738
4 1.000000 0.000000
In [50]: def f(x):
....: return x[0] + x[1]
....:
In [51]: df.apply(f, axis=1)
Out[51]:
0 1.000000
1 0.076619
2 1.000000
3 1.646622
4 1.000000
Example 2: pandas dataframe apply function with multiple arguments
def some_func(row, var1):
return '{0}-{1}-{2}'.format(row['A'], row['B'], var1)
df['C'] = df.apply(some_func(row, var1='DOG'), axis=1)
df
A B C
0 foo x foo-x-DOG
1 bar y bar-y-DOG