Rasterio geometry mask example for Stack Exchange

geometry_mask_example.py

import xarray as xr
import geopandas as gpd
import rasterio

# Open your shapefile and xarray object
ds = raster_mask
gdf = vector_mask

# Select shapefile feature you want to analyse
# and reproject to same CRS as xarray
gdf = gdf.iloc[[0]].to_crs(ds.crs)

# create polygon mask
mask = rasterio.features.geometry_mask(
            gdf.geometry,
            out_shape=ds.geobox.shape,
            transform=ds.geobox.affine,
            all_touched=False,
            invert=False)

mask = xr.DataArray(mask, dims=("y", "x"))

# mask ds with rasterized gdf
ds = ds.where(~mask)

Thanks for the answer @robbibt. Unfortunately, using odc.geo.xr does not add the geobox function onto my xarray dataset. Is it ok to simply open it like this or do I have to link it to old somehow?

EVI_post = xr.open_dataset('~/myfile.nc')