how to apply a function on two columns in pandas code example
Example 1: pandas pass two columns to function
df["Delivery Charges"] = df[["Weight", "Package Size", "Delivery Mode"]].apply(
lambda x : calculate_rate(*x), axis=1)
df["Delivery Charges"] = df.apply(
lambda x : calculate_rate(x["Weight"],
x["Package Size"], x["Delivery Mode"]), axis=1)
Example 2: python pandas apply function to one column
df['a'] = df['a'].apply(lambda x: x + 1)
Example 3: 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