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])