cf_xarray_accessor_for_cf_convention_variable_and_axis_access.py

1import xarray as xr
2import cf_xarray
3import numpy as np
4
5# Create a sample dataset with CF attributes
6ds = xr.Dataset(
7    data_vars={
8        "temp": (("time", "lat", "lon"), np.random.randn(2, 3, 4)),
9    },
10    coords={
11        "time": [np.datetime64("2020-01-01"), np.datetime64("2020-01-02")],
12        "lat": (["lat"], [40, 41, 42], {"units": "degrees_north", "standard_name": "latitude"}),
13        "lon": (["lon"], [10, 11, 12, 13], {"units": "degrees_east", "standard_name": "longitude"}),
14    },
15)
16
17# Access variables using CF standard names or axis names
18print("Latitude values via CF accessor:")
19print(ds.cf["latitude"])
20
21# Select data using CF axis names (X, Y, Z, T)
22subset = ds.cf.sel(T="2020-01-01", Y=slice(40, 41))
23print("\nSubset data dimensions:")
24print(subset.dims)
25
26# Identify which coordinate is the 'X' axis
27print("\nLongitude (X) name:")
28print(ds.cf.axes["X"])