numpy中where函數的用法

numpy.where(condition,x,y)——若滿足condition，輸出x，否則輸出y。

舉例：

例1：一維數組

>>> a = np.arange(10) >>> a array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) >>> np.where(a < 5, a, 10*a) array([ 0, 1, 2, 3, 4, 50, 60, 70, 80, 90])

例2：多維數組

>>> np.where([[True, False], [True, True]], ... [[1, 2], [3, 4]], ... [[9, 8], [7, 6]]) array([[1, 8],[3, 4]])

這個例子解釋一下：條件是

[True, False], [True, True]

也就是說，第一個元素認為是true（真的），則應選擇真值表 [1, 2], [3, 4] 中對應的第一個元素——1。第二個元素認為是false（假的），則應選擇假值表[9, 8], [7, 6]中對應的第二個元素，即為2。第三和四個元素認為是真的，在真值表[1, 2], [3, 4]中選擇對應的第三、四個元素——即為3，4。

例3：

>>> x, y = np.ogrid[:3, :4] >>> np.where(x < y, x, 10 + y) # both x and 10+y are broadcast array([[10, 0, 0, 0],[10, 11, 1, 1],[10, 11, 12, 2]])

例4：

>>> a = np.array([[0, 1, 2], ... [0, 2, 4], ... [0, 3, 6]]) >>> np.where(a < 4, a, -1) # -1 is broadcast array([[ 0, 1, 2],[ 0, 2, -1],[ 0, 3, -1]])

END

------------------------------------

官網鏈接https://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html

總結

以上是生活随笔為你收集整理的numpy中where函数的用法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。