извлечение главной диагонали разреженной матрицы в 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)