INNER CODE UNIT · Python

m

ssebastianmag/hydrogen-wavefunctions · interactive_plot/interactive_plot_util.py:197

        m = max(min(m, m_max), -m_max)

        # Compute Wave Function
        extent_a_mu = 2.5 * n ** 2
        wf = WaveFunction(n=n, l=l, m=m, extent_a_mu=extent_a_mu, grid_points=resolution)
        try:
            P, extent = compute_P_and_extent(wf)
        except MemoryError as e:
            # Specify error
            raise MemoryError(
                f'Encountered MemoryError. Likely caused by too much resolution. Try reducing resolution. Error: {e}')

        x_min, x_max, z_min, z_max = extent
        # Plot
        fig = go.Figure(
            data=[go.Heatmap(
                z=P,
                x=np.linspace(x_min, x_max, P.shape[1]),

View source record →