给定 scipy.sparse.csr.csr_matrix ，是否有快速返回对角线元素的方法？
scipy.sparse.csr.csr_matrix
我想这样做的原因是计算 inv(D) A ，其中 D 是对角线矩阵，其对角线条目与 A 一致（ A 是我的稀疏矩阵，保证在对角线上具有非零值） .
inv(D) A
D
A
使用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)
1 回答
使用csr_matrix.diagonal()：
例：