Ejemplo n.º 1
0
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")