NumPy argwhere() 函数用法详解 - 查找满足条件的元素索引

NumPy 库中的 np.argwhere() 函数用于返回满足条件的元素的索引。它返回一个包含满足条件的元素索引的二维数组。

使用 np.argwhere() 的一般语法如下：

np.argwhere(condition)

其中，'condition' 是一个用于筛选元素的条件。返回的结果是一个二维数组，每一行表示一个满足条件的元素的索引。

例如，假设有一个一维数组 arr=[1, 2, 3, 4, 5]，我们想找出其中所有大于 2 的元素的索引，可以使用 np.argwhere() 函数：

import numpy as np

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

print(indices)

输出结果为：

[[2]
 [3]
 [4]]

这里的结果表示 arr 中索引为 2、3 和 4 的元素满足条件 arr > 2。