def divide_diagonal_by_2(CHI0, div_fact = 2.): CHI = CHI0.copy(); CHI[np.diag_indices_from(CHI)] /= div_fact return CHI