numpyで要素のインデックスを検索する方法は何ですか。

NumPyでは、np.where()関数を使って要素のインデックスを見つけることができます。指定した条件を満たす要素のインデックスを返すnp.where()関数の例を以下に示します。

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])

# 查找元素3的索引
index = np.where(arr == 3)
print(index)  # 输出 (array([2]),)

# 查找元素大于4的索引
index = np.where(arr > 4)
print(index)  # 输出 (array([4, 5]),)

# 查找元素不等于2的索引
index = np.where(arr != 2)
print(index)  # 输出 (array([0, 2, 3, 4, 5]),)

np.where()関数の注意点ですが、条件に合う要素のインデックス配列を含むタプルを返します。インデックス配列を取得したい場合は、index[0]を使用できます。