熊貓添加列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'])