糟糕白猫

文章
7
资源
0
加入时间
3年0月21天

python numpy中nonzero()的用法

nonzero函数返回非零元素的目录。返回值为元组, 两个值分别为两个维度, 包含了相应维度上非零元素的目录值。   import numpy as npA = np.mat([1,1,0,1,0,1,0,0,1])x = A.nonzero()  #取出矩阵中的非零元素的坐标print xprint A[x],'\n'    #取出矩阵中的非零元素