如何根据相邻列中的值将lambda函数应用于特定列

Question

我正在尝试将lambda函数应用于pandas数据帧。我的问题是，如何使用if语句根据b列中的值将lambda函数应用于a列。

A B C
2 5 7
4 5 9
6 7 9

df['B'].apply(lambda x: x+3 if x<(#the value in column C) else x)

Answer 1

您需要使用axis=1在数据帧上调用apply，而不是在B列上调用：

>>> df.apply(lambda x: x['B']+3 if x['B']<x['C'] else x['B'], axis=1)
0     8
1     8
2    10
dtype: int64

但是，更高效(更快(的方法是这样做：

>>> df['B'] + df['B'].lt(df['C']) * 3
0     8
1     8
2    10
dtype: int64