Skip to content

Instantly share code, notes, and snippets.

View aidenrobert's full-sized avatar

Aiden Robert Jönsson aidenrobert

View GitHub Profile
@rabernat
rabernat / detrend_xarray.py
Last active November 19, 2023 05:47
Detrend xarray DataArrays using polyfit
import xarray as xr
def detrend_dim(da, dim, deg=1):
# detrend along a single dimension
p = da.polyfit(dim=dim, deg=deg)
fit = xr.polyval(dim, p.polyfit_coefficients)
return da - fit
def detrend(da, dims, deg=1):
# detrend along multiple dimensions