無効なscipyスパースcsr行列行のインデックスを取得

Question

nullではないscipy.sparse.csr_matrixの行のインデックスを取得したいと考えています。例えば：

A = [ 0 0 0 0 0 1 
     0 0 0 0 0 0 
     1 1 0 0 0 0 ] 

所望の出力：

indices = [0, 2] 

Answer 1

コード：

from scipy.sparse import find, csr_matrix 
import numpy as np 

A = csr_matrix(np.array([[0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0], [1, 1, 0, 0, 0, 0]])) 
print(A.todense()) 

nnz_per_row = A.getnnz(axis=1) 
result = np.where(nnz_per_row > 0)[0] 

print(result) 

出力：

[[0 0 0 0 0 1] 
[0 0 0 0 0 0] 
[1 1 0 0 0 0]] 
[0 2]