nonzero函數(shù)返回非零元素的目錄。
返回值為元組, 兩個值分別為兩個維度, 包含了相應(yīng)維度上非零元素的目錄值。
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 #輸出是一個元組,兩個維度。一一對應(yīng), #返回非零元素在矩陣中的位置,前一個列表存放非零行坐標,后一個列表存放非零元素列坐標 #(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])) #放下來一一對應(yīng),即 #(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()等價
以上這篇淺談python numpy中nonzero()的用法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持VEVB武林網(wǎng)。
新聞熱點
疑難解答
圖片精選