suboptimumg.plotting.utils#

class suboptimumg.plotting.utils.DataType3D(*values)[source]#

Bases: Enum

GridInput = 'GridInput'#
ScatterInput = 'ScatterInput'#
suboptimumg.plotting.utils.prepare_smooth_data_2D(x_list, y_list, smoothing_config)[source]#

Helper function to prepare smoothed data for plots.

Parameters:

  • x_list (numpy.ndarray) – Numpy array of x coordinates

  • y_list (numpy.ndarray) – Numpy array of y coordinates

  • smoothing_config (SmoothingConfig) – SmoothingConfig object

Returns:

  • x_dense (numpy.ndarray) – Smoothed/interpolated x coordinates

  • y_dense (numpy.ndarray) – Smoothed/interpolated y coordinates

suboptimumg.plotting.utils.prepare_smooth_data_3D(x_list, y_list, z_list, smoothing_config=None)[source]#

Helper function to prepare smoothed data for 3D grid plots.

Requires grid-based input data.

Parameters:

  • x_list (numpy.ndarray) – 1D array of x coordinates defining the grid x-axis

  • y_list (numpy.ndarray) – 1D array of y coordinates defining the grid y-axis

  • z_list (numpy.ndarray) – 2D array of z values with shape (len(x_list), len(y_list))

  • smoothing_config (SmoothingConfig, optional) – Smoothing configuration object

Returns:

  • x_interp (numpy.ndarray) – Interpolated x coordinates

  • y_interp (numpy.ndarray) – Interpolated y coordinates

  • z_interp (numpy.ndarray) – Smoothed and interpolated z values

suboptimumg.plotting.utils.prepare_smooth_data_3D_scatter(x_list, y_list, z_list, smoothing_config=None)[source]#

Helper function to interpolate scattered 3D data onto a regular grid.

Requires scatter-based input data (1D x, y, z arrays).

Parameters:

  • x_list (numpy.ndarray) – 1D array of x coordinates

  • y_list (numpy.ndarray) – 1D array of y coordinates

  • z_list (numpy.ndarray) – 1D array of z values

  • smoothing_config (SmoothingConfig, optional) – SmoothingConfig object (uses DEFAULT_SMOOTHING_CONFIG if None)

Returns:

  • x_interp (numpy.ndarray) – Regular grid x coordinates

  • y_interp (numpy.ndarray) – Regular grid y coordinates

  • z_interp (numpy.ndarray) – Interpolated z values on regular grid suitable for surface/contour plots