python中numpy.nonzero

nonzero：返回非零元素的索引。
返回数组的元组，每个数组对应一个维度，其中包含该维度中非零元素的索引。

# 一维
a = [1, 0, 5]
b = nonzero(a)
print(b)

# 二维
a = array([[0, 0, 1], [2, 0, 0], [0, 6, 0]])
b = nonzero(a)
print(b)

# 三维
a = array([[[2, 0], [1, 0]], [[0, 1], [0, 0]], [[0, 0], [0, 5]]])
b = nonzero(a)
print(b)