如何提取稀疏矩阵的主对角线？

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)