在Numpy数组中查找模式

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
condition = arr > 2
result = np.where(condition)
print(result)

(array([2, 3, 4], dtype=int64),)

arr = np.array([1, 2, 3, 4, 5])
condition = arr > 2
result = np.argwhere(condition)
print(result)

[[2]
 [3]
 [4]]

arr = np.array([1, 2, 3, 4, 5])
result = np.nonzero(arr > 2)
print(result)

(array([2, 3, 4], dtype=int64),)

arr = np.array([1, 2, 3, 4, 5])
condition = arr > 2
result = np.extract(condition, arr)
print(result)

[3 4 5]