熊猫添加列if条件

Question

嗨我不知道如何添加一个新的列z到我现有的df，其中z = x如果x不是0 else y。

例如：

Name x y |z (new col) 
A  1 5 |1 
B  2 6 |2 
C  0 3 |3 
D  0 0 |0 

是否np.where适用于这种情况？

谢谢！

Answer 1

你可以试试：

df['z'] = np.where(df['x']!=0,df['x'],df['y'])