浅谈python numpy中nonzero()的用法
nonzero函数返回非零元素的目录。
返回值为元组,两个值分别为两个维度,包含了相应维度上非零元素的目录值。
importnumpyasnp A=np.mat([[0,1,2,3,4,3,2,1,0],[0,1,2,3,4,5,6,7,0]]) x=A.nonzero()#取出矩阵中的非零元素的坐标 printx#输出是一个元组,两个维度。一一对应, #返回非零元素在矩阵中的位置,前一个列表存放非零行坐标,后一个列表存放非零元素列坐标 #(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的 printA[x],'\n'#取出矩阵中的非零元素 #[[12343211234567]] B=np.mat([[1,1,0,1,0,1,0,0,1],[0,1,1,0,0,0,1,1,1]]) printnp.nonzero(B)#与B.nonzero()等价
以上这篇浅谈pythonnumpy中nonzero()的用法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。