SciPy

scipy.sparse.csc_matrix.diagonal

csc_matrix.diagonal()[source]

Returns the main diagonal of the matrix