np.nonzero有什么用

`np.nonzero` 函数用于返回数组中非零元素的索引。具体来说，它返回一个元组，其中包含每个轴上非零元素的索引。这对于查找和处理数组中的非零元素非常有用。

举个例子，假设有一个数组 `arr = np.array([0, 2, 3, 0, 0, 1])`，我们想要找到其中非零元素的索引。可以使用 `np.nonzero(arr)` 函数来实现：

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

这里返回了一个包含一个元组的元组。元组中的第一个元素表示非零元素在原始数组中的索引。在这个例子中，数组中的非零元素是 `2`、`3` 和 `1`，它们在原始数组中的索引分别是 `1`、`2` 和 `5`。

使用 `np.nonzero` 函数，我们可以轻松地查找数组中的非零元素，并在处理数组时使用这些索引。