FeatureSelectionPlot#

class chemotools.plotting.FeatureSelectionPlot(x: ndarray, y: ndarray, support: ndarray, *, selection_color: str = 'red', selection_alpha: float = 0.2, **kwargs: Any)[源代码]

基类:SpectraPlot

Plot class for visualizing feature selection on spectral data.

This class extends SpectraPlot to highlight excluded features using colored vertical spans.

参数:
  • x (np.ndarray) -- X-axis data (e.g., wavelengths, wavenumbers).

  • y (np.ndarray) -- Y-axis data (e.g., spectra intensities).

  • support (np.ndarray) -- Boolean mask indicating selected features (True means selected). Must have same length as x.

  • selection_color (str, optional) -- Color to use for highlighting excluded features. Default is "red".

  • selection_alpha (float, optional) -- Transparency of the selection highlight. Default is 0.2.

  • **kwargs (Any) -- Additional arguments passed to SpectraPlot.__init__.

Attributes

color_by

is_categorical

colormap

colorbar_label

show(figsize=(12, 4), title=None, xlabel='X-axis', ylabel='Y-axis', xlim=None, ylim=None, **kwargs) Figure[源代码]

Show the spectra plot with given figure size and labels. The excluded features are highlighted with red bars.

参数:
  • figsize (tuple, optional) -- Figure size as (width, height) in inches. Default is (12, 4).

  • title (str, optional) -- Title for the plot. If None, a default title is generated.

  • xlabel (str, optional) -- X-axis label. Default is "X-axis".

  • ylabel (str, optional) -- Y-axis label. Default is "Y-axis".

  • xlim (tuple, optional) -- X-axis limits as (xmin, xmax). Default is None (auto).

  • ylim (tuple, optional) -- Y-axis limits as (ymin, ymax). Default is None (auto).

  • **kwargs (Any) -- Additional keyword arguments passed to the plot function.

返回:

The matplotlib Figure object containing the plot.

返回类型:

Figure

render(ax: Axes | None = None, *, xlabel: str | None = None, ylabel: str | None = None, xlim: tuple[float, float] | None = None, ylim: tuple[float, float] | None = None, **kwargs: Any) tuple[Figure, Axes][源代码]

Render the plot on the given axes or create new ones.

参数:
  • ax (Axes, optional) -- Matplotlib axes to plot on. If None, creates new figure and axes.

  • xlabel (str, optional) -- X-axis label. Default is "X-axis".

  • ylabel (str, optional) -- Y-axis label. Default is "Y-axis".

  • xlim (tuple[float, float], optional) -- X-axis limits as (xmin, xmax). When set without ylim, the y-axis automatically scales to fit the data within the x-range.

  • ylim (tuple[float, float], optional) -- Y-axis limits as (ymin, ymax). When provided, disables automatic y-scaling.

  • **kwargs (Any) -- Additional keyword arguments passed to the plot function.

返回:

  • fig (Figure) -- The matplotlib Figure object.

  • ax (Axes) -- The matplotlib Axes object with the rendered plot.