NumPy 配列内の要素の最初のインデックスを見つけるにはどうすればよいですか?

NumPy 配列内の項目の最初のインデックスの検索

Python リストでは、index() メソッドを使用して最初のインデックスを取得できます。要素の出現。 NumPy は配列に対して同様の機能を提供できますか?

回答

はい、NumPy は配列内の項目の最初のインデックスを見つけるための便利な方法を提供します。 np.where 関数は次のように利用できます。

import numpy as np

array = np.array([1, 2, 3])
item = 2

itemindex = np.where(array == item)

np.where 関数は、項目が見つかった行と列のインデックスを含むタプルを返します。

例

配列が 2 次元で、2 つの場所に項目が含まれている場合、 then:

array[itemindex[0][0]][itemindex[1][0]]  # equals item
array[itemindex[0][1]][itemindex[1][1]]  # also equals item

これにより、最初に出現した要素に基づいて目的の要素に簡単にアクセスできます。

以上がNumPy 配列内の要素の最初のインデックスを見つけるにはどうすればよいですか?の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。