Python中如何使用numpy.where()函數

小編給大家分享一下Python中如何使用numpy.where()函數，希望大家閱讀完這篇文章后大所收獲，下面讓我們一起去探討吧！

numpy.where(condition [，x，y])函數返回滿足給定條件的輸入數組中元素的索引。

參數:condition : When True, yieldx, otherwise yield y.x,y: Values from which to choose. x, y and

condition need to be broadcastable to some shape.

返回:out : [ndarray or tuple of ndarrays] If both x and y are specified, the output array contains elements of x where condition is True, and elements from y elsewhere.

If only condition is given, return the tuple condition.nonzero(), the indices where condition is True.

代碼1：

# Python program explaining  
     # where() function  
  
     import numpy as np 
  
     np.where([[True, False], [True, True]], 
         [[1, 2], [3, 4]], [[5, 6], [7, 8]])

輸出：

array([[1, 6],
       [3, 4]])

代碼2：

# Python program explaining  
     # where() function  
  
     import numpy as np 
  
     # a is an array of integers. 
     a = np.array([[1, 2, 3], [4, 5, 6]]) 
  
     print(a) 
  
     print ('Indices of elements <4') 
  
     b = np.where(a<4) 
     print(b) 
  
     print("Elements which are <4") 
     print(a[b])

輸出：

[[1 2 3]
 [4 5 6]]

Indices of elements <4
(array([0, 0, 0], dtype=int64), array([0, 1, 2], dtype=int64))

Elements which are <4
array([1, 2, 3])

看完了這篇文章，相信你對Python中如何使用numpy.where()函數有了一定的了解，想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！