np.nonzero参数详解

np.nonzero是NumPy库中的一个函数，它用于返回数组中非零元素的索引。它接受一个数组作为参数，并返回一个包含非零元素索引的元组。在这个元组中，每个维度都有一个数组，分别包含该维度上非零元素的索引。

下面是使用np.nonzero的一个示例：

import numpy as np

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

indices = np.nonzero(arr)
print(indices)

输出结果为：

(array([0, 1, 1]), array([1, 0, 2]))

在这个例子中，输入数组arr有两个非零元素，分别是1和3。使用np.nonzero返回的元组中的第一个数组表示非零元素所在的行索引，第二个数组表示非零元素所在的列索引。

需要注意的是，np.nonzero返回的索引数组的长度等于输入数组中非零元素的个数。如果输入数组是多维的，则每个维度都有对应的索引数组。

希望对你有所帮助！如果还有其他问题，请继续提问。