获取矩阵内非零元素坐标

import numpy as np

a = np.array([[1,1,0], [0,1,0],[0,0,1]])
print(a)
b = a==1
print(b)
nonb = b.nonzero()
print("nonbx", nonb[0])
print("nonby", nonb[1])

c = np.zeros_like(a)
print(c)
c[b] = 1
print(c)