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，不知道对不对