在 NumPy 陣列中查詢元素的第一個索引
Manav Narula
2023年1月30日
2021年4月29日
-
使用
where()
函式在 NumPy 陣列中查詢元素的第一個索引 -
使用
nonzero()
函式在 NumPy 陣列中查詢元素的第一個索引 -
使用
argmax()
函式在 numpy 陣列中查詢元素的第一個索引 -
使用
index()
函式在 NumPy 陣列中查詢元素的第一個索引
在本教程中,我們將討論如何在 numpy 陣列中查詢元素的第一個索引。
使用 where()
函式在 NumPy 陣列中查詢元素的第一個索引
numpy 模組中的 where()
函式用於返回一個陣列,該陣列包含滿足某些條件的元素的索引。條件在函式中指定。
我們可以使用它來查詢陣列中特定值的第一個索引,如下所示。
a = np.array([7,8,9,5,2,1,5,6,1])
print(np.where(a==1)[0][0])
輸出:
5
使用 nonzero()
函式在 NumPy 陣列中查詢元素的第一個索引
nonzero()
函式返回 numpy 陣列中所有非零元素的索引。它為多維陣列返回多個陣列的元組。
與 where()
函式類似,我們也可以指定條件,以便它也可以返回特定元素的位置。
例如,
a = np.array([7,8,9,5,2,1,5,6,1])
print(np.nonzero(a==1)[0][0])
輸出:
5
為了最基本的目的,where()
和 nonzero()
函式看起來很相似。where()
函式的區別在於,當你希望從陣列 a
中選取某些條件為 True
時的元素,以及從陣列 b
中選取該條件為 False
時的元素。
使用 argmax()
函式在 numpy 陣列中查詢元素的第一個索引
argmax()
查詢陣列中最大元素的索引。我們可以在函式中指定相等條件,並找到所需元素的索引。
例如,
a = np.array([7,8,9,5,2,1,5,6,1])
print(np.argmax(a==1))
輸出:
5
使用 index()
函式在 NumPy 陣列中查詢元素的第一個索引
在這種方法中,我們將首先使用 tolist()
函式將陣列轉換為列表。然後,我們將使用 index()
函式,該函式將返回指定元素的位置。
以下程式碼實現了這一點。
a = np.array([7,8,9,5,2,1,5,6,1])
print(a.tolist().index(1))
輸出:
5
Author: Manav Narula
Manav is a IT Professional who has a lot of experience as a core developer in many live projects. He is an avid learner who enjoys learning new things and sharing his findings whenever possible.
LinkedIn