SciPy

scipy.sparse.dok_matrix.diagonal

dok_matrix.diagonal()[source]

Returns the main diagonal of the matrix