有没有办法在DataFrame中使用apply和lambda时应用条件？

import pandas as pd

# 创建一个示例DataFrame
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50]
}
df = pd.DataFrame(data)

# 使用apply和lambda函数应用条件
df['C'] = df.apply(lambda row: row['A'] + row['B'] if row['A'] > 2 else row['A'] * row['B'], axis=1)

print(df)

   A   B   C
0  1  10   1
1  2  20   4
2  3  30  63
3  4  40  84
4  5  50 125