desc.integrals.Bounce2D.plot_angle_spectrum

static Bounce2D.plot_angle_spectrum(angle, l, *, truncate=0, norm=<matplotlib.colors.LogNorm object>, h_ax_numticks=None, v_ax_numticks=None, **kwargs)Source

Plot frequency spectrum of the given stream map.

Parameters:

  • angle (jnp.ndarray) – Shape (num ρ, X, Y). Angle returned by Bounce2D.angle.

  • l (int) – Index into first axis of angle.

  • truncate (int) – Index at which to truncate any Chebyshev series. This will remove aliasing error at the shortest wavelengths where the signal to noise ratio is lowest. The default value is zero which is interpreted as no truncation.

  • norm (str) – The normalization method used for the color scale. See https://matplotlib.org/stable/api/_as_gen/matplotlib.pyplot.imshow.html. Default is logarithmic scale with cutoff at 1e-7.

  • h_ax_numticks (int) – If given, labels at most h_ax_numticks marks on the horizontal axis.

  • v_ax_numticks (int) – If given, labels at most v_ax_numticks marks on the vertical axis.

  • kwargs – Keyword arguments to pass to matplotlib.

Returns:

fig, ax – Matplotlib (fig, ax) tuple.