nonzero函数返回非零元素的目录。

返回值为元组, 两个值分别为两个维度, 包含了相应维度上非零元素的目录值。
import numpy as np
A = np.mat([[0,1,2,3,4,3,2,1,0],[0,1,2,3,4,5,6,7,0]])
x = A.nonzero() #取出矩阵中的非零元素的坐标
print x #输出是一个元组,两个维度。一一对应,
#返回非零元素在矩阵中的位置,前一个列表存放非零行坐标,后一个列表存放非零元素列坐标
#(array([0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]), array([1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 6, 7]))
#放下来一一对应,即
#(array([0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]),
#array([1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 6, 7]))
# (0,1) (0,2).............................. (1,6) (1,7) 坐标上的元素都是非0的
print A[x],'\n' #取出矩阵中的非零元素
#[[1 2 3 4 3 2 1 1 2 3 4 5 6 7]]
B = np.mat([[1,1,0,1,0,1,0,0,1],[0,1,1,0,0,0,1,1,1]])
print np.nonzero(B) # 与B.nonzero()等价 另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。