nonzero

Posted 2021-02-08 xiashiwendao

在python的numpy里面这个函数的意义是返回参数数组中不为0的元素的索引（indics）。

from numpy import array

1 from numpy import nonzero 
2 x=array([[1,0,0], [0,2,0], [1,1,0]]) 
3 print(x) 
4 nonzero(x)

 

output：

[[1 0 0] [0 2 0] [1 1 0]] (array([0, 1, 2, 2], dtype=int64), array([0, 1, 0, 1], dtype=int64))

这个是要把两行捏在一起来看：

[0,1,2,2]

[0,1,0,1]

代表作为二维数组[0,0]，[1,1]，[2,0]，[2,1]四个位置的元素不为0，也就说返回的一个数组是行信息，第二个数组对应的列信息，组合在一起就是一个（x,y）坐标。