desc.integrals.Bounce2D.plot_angle_spectrum

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

Plot frequency spectrum of the given inverse stream map.

Parameters:

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

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

  • 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.