Récupère les index des lignes de matrices csr non nulles scipy sparse

Question

Je veux obtenir les index des lignes d'un scipy.sparse.csr_matrix qui ne sont pas nulles. Par exemple:

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

sortie désiré:

indices = [0, 2] 

Answer 1

code:

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) 

sortie:

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