Skip to content

Instantly share code, notes, and snippets.

@davipatti

davipatti/assign_coords.py

Last active August 2, 2024 18:41
Show Gist options
  • Save davipatti/c3a160c2e54c67d903acd62e59c8fe16 to your computer and use it in GitHub Desktop.
Save davipatti/c3a160c2e54c67d903acd62e59c8fe16 to your computer and use it in GitHub Desktop.
Download ZIP
[Rename coordinates using assign_coords] How to rename coordinates in an xarray dataset #pymc #arviz #xarray
Raw
assign_coords.py
def rename_coordinates(dataset: "xarray.Dataset", dim: str, mapping: dict[str, str]) -> "xarray.Dataset":
return dataset.assign_coords(**{dim: [mapping[k] for k in dataset.coords[dim].values]})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment