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

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)