np.nonzero()

np.nonzero() 函数返回输入数组中非零元素的索引。

import numpy as np

a = np.array([[30, 40, 0], [0, 20, 10], [50, 0, 60]])
print(a)
print(np.nonzero(a))

输出结果：

[[30 40  0]
 [ 0 20 10]
 [50  0 60]]


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