在numpy中，我們可以使用argwhere函數來獲取數組中符合條件的元素的索引位置。下面我們將從多個方面詳細闡述如何使用argwhere函數。
一、基本使用方法
argwhere函數接受一個條件作為輸入，並返回一個包含符合條件的元素索引的數組，其中每個元素是一個n維數組，其形狀與輸入數組相同。
import numpy as np
arr = np.array([[0, 1, 3], [1, 2, 3], [4, 5, 6]])
indices = np.argwhere(arr > 2)
print(indices)
在上述示例中，我們創建了一個3×3的numpy數組，然後使用argwhere函數找出大於2的元素的索引位置，並將結果列印出來。
二、多個條件的使用方法
argwhere函數還可以同時接受多個條件作為輸入，並返回一個包含符合這些條件的元素索引的數組。
indices = np.argwhere((arr > 2) & (arr % 2 == 0))
print(indices)
在上述示例中，我們使用了邏輯運算符&來組合兩個條件，找出大於2並且為偶數的元素的索引位置。
三、使用條件函數作為參數
argwhere函數還可以接受一個函數作為參數，該函數輸入數組的每個元素，並返回一個布爾值。
def is_odd(num):
return num % 2 == 1
indices = np.argwhere(is_odd(arr))
print(indices)
在上述示例中，我們創建了一個名為is_odd的函數來判斷一個數是否為奇數，然後使用該函數作為argwhere函數的參數，找出數組中所有的奇數元素的索引位置。
四、使用argwhere的返回值進行切片
由於argwhere函數返回一個n維數組，我們可以使用這個數組來對原數組進行切片。
arr[indices[:, 0], indices[:, 1]]
# Output: array([3, 5])
在上述示例中，我們使用argwhere函數返回的索引位置進行切片，在原數組中返回所有找到的元素。
五、總結
argwhere函數是numpy中非常有用的一個函數，可以在很多場景中幫助我們快速獲取符合條件的元素的索引位置。在實際應用中，我們可以根據需要設置不同的參數，以獲取特定條件下的元素索引位置，並對原數組進行切片等操作。
原創文章，作者：EBFTB，如若轉載，請註明出處：https://www.506064.com/zh-tw/n/368508.html
微信掃一掃 
支付寶掃一掃 