np.argwhere的用法

`np.argwhere`函数返回数组中非零元素的索引。

具体用法如下：

import numpy as np

arr = np.array([[0, 1, 0],
                [1, 0, 1],
                [0, 1, 0]])

indices = np.argwhere(arr != 0)

print(indices)

输出：

[[0 1]
 [1 0]
 [1 2]
 [2 1]]

在这个例子中，我们首先创建了一个二维的numpy数组`arr`。然后，我们使用`np.argwhere`函数查找`arr`中非零元素的索引。在这个例子中，返回的结果是一个二维的numpy数组，其中每一行记录着一个非零元素的索引。

需要注意的是，`np.argwhere`返回的索引数组的形状与原始数组的维数相同。在上面的例子中，`arr`是一个二维数组，因此返回的索引数组也是一个二维数组。如果`arr`是一个一维数组，则返回的索引数组也是一个二维数组，其中每一行只有一个元素。