在scipy.sparse python中提取稀疏矩阵的主对角线

Question

如何提取稀疏矩阵的主对角线？矩阵在scipy.sparse中创建.我想要等效的np.diagonal()但是对于稀疏矩阵.

Answer 1

稀疏矩阵有一个diagonal方法:

M.diagonal()

http://docs.scipy.org/doc/scipy/reference/generated/scipy.sparse.csc_matrix.diagonal.html

numpy对角线更强大,允许您指定对角线

M.A.diagonal(2)