SciPy

scipy.sparse.csr_matrix.diagonal

csr_matrix.diagonal()[source]

Returns the main diagonal of the matrix