python中numpy.nonzero()用法

举例说明：

-------->>> x = np.eye(3)>>> xarray([[ 1.,  0.,  0.],       [ 0.,  1.,  0.],       [ 0.,  0.,  1.]])>>> np.nonzero(x)(array([0, 1, 2]), array([0, 1, 2]))

返回的是一个2维元组，表示矩阵中非零元素的下标，其中第一维表示的是矩阵中非零元素的行下标，第二维是非零元素的列下标。

即非零元素下标为：

[0,0]

[1,1]

[2,2]

扩展部分：

如果想直接得到矩阵中非零元素的下标，进行如下操作

>>> np.transpose(np.nonzero(x))array([[0, 0],       [1, 1],       [2, 2]])

如果想要直接获得矩阵中的非零元素值，可进行如下操作：

>>> x[np.nonzero(x)]array([ 1.,  1.,  1.])