Python Numpy 数组查找非零元素索引

可以使用 numpy 库中的 nonzero 函数来找到数组中非零的位置索引。

以下是一个示例代码：

import numpy as np

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

print(nonzero_indices)

输出结果为：

(array([1, 3, 4, 6, 7]),)

这表示在数组 arr 中，非零的位置索引分别为 1, 3, 4, 6, 7。