Jak mogę podać nazwy wierszy i kolumn dla csr_matrix Scipy?

Question

Nie wiem, czy to możliwe, i jest to prawdopodobnie naiwne pytanie, ale jak ustawić odpowiednik rownames() R i colnames() na scipy.sparse.csr.csr_matrix?

Widziałem, że my_matrix.dtype.names tu nie pracuje, i nie mogę znaleźć żadnej „indeks” equivalend dla takiej rzadkiej matrycy ...

Ponadto pandas.sparse.* nie jest opcją tutaj, z powodu jakiegoś problemu otwartego ...

Dziękuję bardzo za pomoc,

Answer 1

Musisz zachować nazwy oddzielnie, jak żaden z nielicznych formatów scipy obsługują nazwie indeksowania. Może to wyglądać następująco:

foo = csr_matrix(...) 
row_names = np.array(...) 
col_names = np.array(...) 

# index by name: 
row_idx, = np.where(row_names == "my row") 
col_idx, = np.where(col_names == "my col") 
foo[row_idx, col_idx]