Create xarray mask from vector

mask_xarray.py

import xarray as xr
import geopandas as gpd

from dea_tools.spatial import xr_rasterize

#read shapefile
mask_vector = gpd.read_file('mask.shp')

#import xarray ds
ds = xr.open_dataset('testing.nc')

#create mask using ds as template
mask = xr_rasterize(gdf=mask_vector, ds)

#mask the ds using the rasterized shape
ds = ds.where(mask)