Python：只有非零條目的數組中的所有行的索引

Question

在numpy數組中，如何找到只有非零條目的所有行的索引。例如，陣列中：

A = np.array([[ 1, 0, 5], 
       [25, 2, 0], 
       [ 7, 8, 9], 
       [ 0, 0, 4], 
       [11, 14, 15]]) 

我想有[2,4]作爲輸出，由於行2和4是其中所有條目是非零的唯一的行。

目前，我使用

B = A[np.all(A != 0, axis=1)] 

獲得其中至少有一個零的所有行已經被丟棄的數組。但我需要找到指數（即2和4）。

Answer 1

你的方法應該以一點變化如下工作：

np.where(np.all(A != 0, axis=1))[0].tolist() 
Out[284]: [2, 4] 

Answer 2

In [1]: A = np.array([[ 1, 0, 5], 
         [25, 2, 0], 
         [ 7, 8, 9], 
         [ 0, 0, 4], 
         [11, 14, 15]]) 

In [2]: Indx1 = np.all(A != 0, axis=1) 
     print Indx1 
Out[2]: Indx1 = [False False True False True] 

In [3]: Indx2 = np.where(Indx1==True) 
     print Indx2 
Out[3]: (array([2, 4]),) 

In [4]: Indx = A[Indx2[0]] 
     print Indx 
Out[4]: [2 4]