a=np.array([[0,1],[0,1]]) print(a) b=np.where(a!=0) print(b) c=list(map(list, zip(*b))) print(c)
输出如下:
[[0 1]
[0 1]]
(array([0, 1], dtype=int64), array([1, 1], dtype=int64))
[[0, 1], [1, 1]]
a=np.array([[0,1],[0,1]]) print(a) b=np.where(a!=0) print(b) c=list(map(list, zip(*b))) print(c)
输出如下:
[[0 1]
[0 1]]
(array([0, 1], dtype=int64), array([1, 1], dtype=int64))
[[0, 1], [1, 1]]