arpes.analysis.filters.savgol_filter_multi

arpes.analysis.filters.savgol_filter_multi(data, axis_params, deriv=0, mode='interp', cval=0.0, **kwargs)

Apply Savitzky-Golay filter to an xarray.DataArray along multiple dimensions.

Parameters:

• data (xr.DataArray) – The input DataArray.

• axis_params (dict) – Dictionary mapping axis names to dicts of filter parameters, e.g., {“time”: (11, 3), …} # the first item in tuple is window-length, the second is polyorder. # (1, 0) is no smoothing.

• mode (str) – Mode for savgol_filter (default: “interp”).

• deriv – the order of the derivative to compute (default = 0 means only smoothing)

• cval (float) – Constant value used if mode == ‘constant’.

• **kwargs – Additional keyword arguments passed to savgol_filter.

Returns:

The filtered DataArray.

Return type:

xr.DataArray