scipyスパース行列の対角要素を返す

Question

scipy.sparse.csr.csr_matrixを指定すると、対角要素を素早く返すことができますか？私はこれをやりたい

理由はDは対角エントリ（Aが対角線上の非ゼロを持つことが保証私のスパース行列、である）Aに同意する対角行列であるinv(D) Aを計算することです。

Answer 1

使用csr_matrix.diagonal()：

は、行列の主対角返し

例：

>>> import numpy as np 
>>> from scipy.sparse import csr_matrix 
>>> mymat = csr_matrix((4, 4), dtype=np.int8) 
>>> mymat.diagonal() 
array([0, 0, 0, 0], dtype=int8)