Sum the matrix over the given axis. If the axis is None, sum over both rows and columns, returning a scalar.
scipy.sparse.dok_matrix.setdiag
scipy.sparse.dok_matrix.toarray