def sort_key_list(key_list): ke_pair_list = [] for key in key_list: energy = integrals.mat_element(key, key) ke_pair_list.append((energy, key)) ke_pair_list.sort() sorted_key_list = [] for ke in ke_pair_list: sorted_key_list.append(ke[1]) return sorted_key_list
def draw_H(): import integrals import ctrl_panel dim = len(ctrl_panel.key_list) H = [] for key_i in ctrl_panel.key_list: vec = [] for key_j in ctrl_panel.key_list: if key_i != key_j: entry = integrals.mat_element(key_i, key_j, integrals.ref_energy) vec.append(abs(entry)) else: vec.append(0) vec.append(0) H.append(vec) H.append([0] * (dim+1)) plot_once_2D(dim, H)