薩姆2個大熊貓數據幀的列與行條件

Question

 
    A B C 
0 0 -2 1 
1 1 -1 2 
2 2 0 3 
3 3 1 0 
4 4 2 5 

我希望得到一個新的列，使得其拾取元件從乙如果B爲正（> 0），否則拾取選自C

 
    A B C D 
0 0 -2 1 1 
1 1 -1 2 2 
2 2 0 3 3 
3 3 1 0 1 
4 4 2 5 5 

元件

Answer 1

使用布爾掩碼：

In [14]: df[df[['B', 'C']] > 0].max(1) 
Out[14]: 
0 1 
1 2 
2 3 
3 1 
4 5 
dtype: float64 

In [15]: df['D'] = df[df[['B', 'C']] > 0].max(1)