python2个参数的apply函数优化

Question

def checkformat(df):     if df['A'] == 0:         return 0     else:         return np.log(df['A'] / df['B']) df['C'] = df.apply(checkformat, axis=1)请问以上代码，改成apply调用lambda函数怎么用？谢谢！

Answer 1

lambda x: np.log(x['A'] / x['B']) if x['A'] != 0 else 0，不知道对不对