clip an xarray object - getting Type Error

[raybellwaves]

Originally posted at pydata/xarray#4948

Trying to grab data for my home state of Florida from an xarray object but getting a Type Error. Traceback below

Thanks for the feedback. Works now. Code updated.

import xarray as xr
import rioxarray
import geopandas

ds = xr.tutorial.open_dataset("air_temperature")
da = ds.isel(time=0).air
# Convert lon to -180 - 180
da = da.assign_coords(lon=(((da.lon + 180) % 360) - 180))
# Reverse latitude to make it increasing
da = da.reindex(lat=ds.lat[::-1])

# Grab a geopandas dataframe for Florida
url = "https://eric.clst.org/assets/wiki/uploads/Stuff/gz_2010_us_040_00_20m.json"
gdf = geopandas.read_file(url)
gdf = gdf[gdf.NAME == 'Florida']

# Use rioxarray to clip da
rda = da.rio.set_spatial_dims('lon', 'lat')
rda = rda.rio.write_crs("epsg:4326")
da = rda.rio.clip(gdf.geometry, gdf.crs)

Traceback

TypeError                                 Traceback (most recent call last)
<ipython-input-7-fb8331d4e96b> in <module>
----> 1 rda.clip(gdf.geometry, gdf.crs)

~/miniconda3/envs/main/lib/python3.8/site-packages/xarray/core/dataarray.py in func(self, *args, **kwargs)
   2964                 )
   2965                 with np.errstate(all="ignore"):
-> 2966                     da = self.__array_wrap__(f(self.variable.data, *args, **kwargs))
   2967                 if keep_attrs:
   2968                     da.attrs = self.attrs

~/miniconda3/envs/main/lib/python3.8/site-packages/xarray/core/ops.py in func(self, *args, **kwargs)
    228 def _method_wrapper(name):
    229     def func(self, *args, **kwargs):
--> 230         return _call_possibly_missing_method(self, name, args, kwargs)
    231 
    232     func.__name__ = name

~/miniconda3/envs/main/lib/python3.8/site-packages/xarray/core/ops.py in _call_possibly_missing_method(arg, name, args, kwargs)
    214         raise
    215     else:
--> 216         return method(*args, **kwargs)
    217 
    218 

~/miniconda3/envs/main/lib/python3.8/site-packages/numpy/core/_methods.py in _clip(a, min, max, out, casting, **kwargs)
    156             um.maximum, a, min, out=out, casting=casting, **kwargs)
    157     else:
--> 158         return _clip_dep_invoke_with_casting(
    159             um.clip, a, min, max, out=out, casting=casting, **kwargs)
    160 

~/miniconda3/envs/main/lib/python3.8/site-packages/numpy/core/_methods.py in _clip_dep_invoke_with_casting(ufunc, out, casting, *args, **kwargs)
    110     # try to deal with broken casting rules
    111     try:
--> 112         return ufunc(*args, out=out, **kwargs)
    113     except _exceptions._UFuncOutputCastingError as e:
    114         # Numpy 1.17.0, 2019-02-24

~/miniconda3/envs/main/lib/python3.8/site-packages/pandas/core/generic.py in __array_ufunc__(self, ufunc, method, *inputs, **kwargs)
   1933         self, ufunc: Callable, method: str, *inputs: Any, **kwargs: Any
   1934     ):
-> 1935         return arraylike.array_ufunc(self, ufunc, method, *inputs, **kwargs)
   1936 
   1937     # ideally we would define this to avoid the getattr checks, but

~/miniconda3/envs/main/lib/python3.8/site-packages/pandas/core/arraylike.py in array_ufunc(self, ufunc, method, *inputs, **kwargs)
    356         # ufunc(series, ...)
    357         inputs = tuple(extract_array(x, extract_numpy=True) for x in inputs)
--> 358         result = getattr(ufunc, method)(*inputs, **kwargs)
    359     else:
    360         # ufunc(dataframe)

TypeError: '>=' not supported between instances of 'float' and 'MultiPolygon'

[snowman2]

I think that you are missing the .rio part. Should be:

rda.rio.clip(gdf.geometry, gdf.crs)

[raybellwaves]

Thanks. Now getting MissingCRS: CRS not found. Please set the CRS with 'rio.write_crs()'. Data variable: air

----> 1 da.rio.clip(gdf.geometry, gdf.crs)

~/miniconda/envs/main/lib/python3.8/site-packages/rioxarray/raster_array.py in clip(self, geometries, crs, all_touched, drop, invert, from_disk)
    692         """
    693         if self.crs is None:
--> 694             raise MissingCRS(
    695                 "CRS not found. Please set the CRS with 'rio.write_crs()'."
    696                 f"{_get_data_var_message(self._obj)}"

[snowman2]

Sounds like you are on the right path.