Python中的 numpy.argwhere()

numpy.argwhere() 是一个numpy库中的函数，用于查找数组中非零元素的索引。它返回一个数组，其中每行都是一个非零元素的索引。

语法

该函数的语法如下：

numpy.argwhere(arr)

上述语法中，arr是一个输入的数组。

参数

该函数只有一个参数：输入的数组。

返回值

返回一个数组，其中每行都是一个非零元素的索引。

示例

import numpy as np

x = np.arange(6).reshape(2,3)
print("x:\n", x)
print("Nonzero elements of x:\n", np.argwhere(x > 1))

上述代码输出为：

x:
 [[0 1 2]
  [3 4 5]]
 
Nonzero elements of x:
 [[0 2]
  [1 0]
  [1 1]
  [1 2]]

上述代码中，np.argwhere(x > 1) 查找数组 x 中所有大于 1 的元素的索引并返回数组。输出表示有四个大于 1 的元素，分别位于（0,2）、（1,0）、（1,1）和（1,2）位置。

总结

numpy.argwhere() 是一个非常有用的函数，可以用于查找数组中非零元素的索引。当处理大型数组时，这个函数可以帮助您快速找到数组中的非零元素。