返回scipy稀疏矩阵的对角线元素

Question

给定scipy.sparse.csr.csr_matrix，是否有快速的方法来返回对角线上的元素？

我想这样做的原因是为了计算inv(D) A，其中D是对角矩阵，其对角线项同意A（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)