import numpy print(sum(numpy.diag(norm_list)))