Source code for xarray_cube.accessor
import xarray as xr
from astropy.io import fits
from astropy.wcs import WCS
[docs]@xr.register_dataarray_accessor("cube")
class CubeAccessor:
def __init__(self, xarray_obj: xr.DataArray) -> None:
self._obj = xarray_obj
@property
def header(self) -> fits.Header:
"""Convert the header string to a header object."""
return fits.Header.fromstring(self._obj.header.item())
@property
def wcs(self) -> WCS:
"""Convert the header string to a WCS object"""
return WCS(self.header)