def Bound_State_Plotter(input_par, m_n_array): BS, BE = Mod.Bound_State_Reader(input_par) grid = Mod.Make_Grid(input_par["grid_spacing"], input_par["grid_size"], input_par["grid_spacing"]) for m_n in m_n_array: m, n = m_n[0], m_n[1] wave_function = np.zeros(len(grid), dtype=complex) print(n, np.absolute(BE[(m, n)])) for l in range(0, input_par["l_max_bound_state"] + 1): wave_function += BS[(n, l, m)] plt.plot(grid, np.absolute(wave_function)) plt.xlim(0, 50) plt.savefig("WF.png")