使用一些條件

Question

例如替換numpy的數組元素，我有一些陣列等：

>>> x = np.arange(-5, 4).reshape(3, 3) 
>>> x 
array([[-5, -4, -3], 
     [-2, -1, 0], 
     [ 1, 2, 3]]) 

如何可以替換b其是大於a否則它們設置爲0所有元素？

我試圖

np.place(x, lambda y: b if y > a else 0) 

但它沒有發揮出來。

Answer 1

您可以使用numpy.where：

x = np.arange(-5, 4).reshape(3, 3) 
x 
#array([[-5, -4, -3], 
#  [-2, -1, 0], 
#  [ 1, 2, 3]]) 

b = 1; a = 0; 
np.where(x > a, b, 0) 
#array([[0, 0, 0], 
#  [0, 0, 0], 
#  [1, 1, 1]]) 

Answer 2

還不如np.where那樣好，但是你的情況，你可以簡單地乘以「布爾數組」當你與b比較你的數組a你：

>>> x = np.arange(-5, 4).reshape(3, 3) 
>>> a, b = 0, 6 
>>> (x > a) * b 
array([[0, 0, 0], 
     [0, 0, 0], 
     [6, 6, 6]]) 

這是可行的，因爲True相當於1和False到0算術運算。