ENH: xarray grid output

[syedhamidali]

• Implementation of Does PyART have the slice option for the grid objects #1476 (comment)

To streamline the process of extracting a subset from pyart.core.grid or the grid object, I propose to add a few more variables to the xarray dataset that mirrors the grid object. When this xarray dataset is written to a NetCDF file, it becomes equivalent to the original pyart.core.grid object.

How can this be utilized?

import pyart

# Reading a grid from a netcdf file
grid = pyart.io.read_grid("file.nc")

# Converting the grid to an xarray dataset
ds = grid.to_xarray()

# Slicing the dataset based on x and y coordinates
ds = ds.sel(x=slice(5000, 70000), y=slice(5000, 70000), drop=True)

# Saving the sliced dataset to a new netcdf file
ds.to_netcdf("outfile.nc")

Subsequently, it is possible to read the modified dataset using pyart as follows:

# Reading the modified dataset with pyart
grid_small = pyart.io.read_grid("outfile.nc")

This enhances the process of working with grid data, making it more convenient and efficient for users.

[syedhamidali]

Hey @mgrover1,

Just wanted to check in on the status of #1477. I believe we’ve completed the function that allows us to directly write xarray grids using ds.to_netcdf method. This functionality should enable us to seamlessly read these grids back in using Py-ART’s grid reader. Let me know if you have any updates or if there are any additional steps required.

[mgrover1]

@syedhamidali - can you please review the previous comment to ensure that is addressed to fix the failing CI?

[syedhamidali]

@syedhamidali - can you please review the previous comment to ensure that is addressed to fix the failing CI?

Resolved in this commit

[mgrover1]

Fantastic! Thanks @syedhamidali 👍