WDM.code.plotting package

Submodules

WDM.code.plotting.plotting module

WDM.code.plotting.plotting.frequency_domain_plot(wdm: WDM_transform, x: Array) → Tuple[Figure, Axes][source]

Plot the frequency-domain signal.

Parameters:

  • wdm (WDM_transform) – The WDM transform object.

  • x (jnp.ndarray) – Array shape (N,). Input time-domain signal to be plotted.

Returns:

  • fig (matplotlib.figure.Figure) – The matplotlib Figure object.

  • ax (matplotlib.axes.Axes) – The matplotlib Axes object where the wavelets were plotted.

Notes

This function does not call plt.show(). The user is responsible for displaying or saving the plot.

WDM.code.plotting.plotting.time_domain_plot(wdm: WDM_transform, x: Array) → Tuple[Figure, Axes][source]

Plot the time-domain signal.

Parameters:

  • wdm (WDM_transform) – The WDM transform object.

  • x (jnp.ndarray) – Array shape (N,). Input time-domain signal to be plotted.

Returns:

  • fig (matplotlib.figure.Figure) – The matplotlib Figure object.

  • ax (matplotlib.axes.Axes) – The matplotlib Axes object where the wavelets were plotted.

Notes

This function does not call plt.show(). The user is responsible for displaying or saving the plot.

WDM.code.plotting.plotting.time_frequency_plot(wdm: WDM_transform, w: Array, part='abs', scale='linear') → Tuple[Figure, Axes][source]

Plot the time-frequency coefficients of the WDM transform.

Parameters:

  • wdm (WDM_transform) – The WDM transform object.

  • w (jnp.ndarray of shape (Nt, Nf)) – WDM time-frequency coefficients to be plotted.

  • part (str) – Part of the coefficients to plot. Options are ‘abs’ for magnitude, ‘real’, or ‘imag’. Default is ‘abs’. Optional.

  • scale (str) – Scale of the colour axis of the plot. Passed to matplotlib. Options are ‘linear’ or ‘log’. Default is ‘linear’. Logarithmic scale should only be used with part=’abs’ otherwise problems with negative values will occur. Optional.

Returns:

  • fig (matplotlib.figure.Figure) – The matplotlib Figure object.

  • ax (matplotlib.axes.Axes) – The matplotlib Axes object where the wavelets were plotted.

Notes

This function does not call plt.show(). The user is responsible for displaying or saving the plot.