Is your feature request related to a problem?

I think that having a way to run a multidimensional interpolation for filling missing values would be awesome.

The code snippet below create a data and show the problem I am having now. If the data has some orientation, we couldn't simply interpolate dimensions separately.

```python import xarray as xr import numpy as np

n = 30 x = xr.DataArray(np.linspace(0,2np.pi,n),dims=['x']) y = xr.DataArray(np.linspace(0,2np.pi,n),dims=['y']) z = (np.sin(x)*xr.ones_like(y))

mask = xr.DataArray(np.random.randint(0,1+1,(n,n)).astype('bool'),dims=['x','y'])

kw = dict(add_colorbar=False)

fig,ax = plt.subplots(1,3,figsize=(11,3)) z.plot(ax=ax[0],kw) z.where(mask).plot(ax=ax[1],kw) z.where(mask).interpolate_na('x').plot(ax=ax[2],**kw) ```

I tried to use advanced interpolation for that, but it doesn't look like the best solution.

```python zs = z.where(mask).stack(k=['x','y']) zs = zs.where(np.isnan(zs),drop=True) xi,yi = zs.k.x.drop('k'),zs.k.y.drop('k') zi = z.interp(x=xi,y=yi)

fig,ax = plt.subplots() z.where(mask).plot(ax=ax,kw) ax.scatter(xi,yi,c=zi,kw,linewidth=1,edgecolor='k') ``` returns

Describe the solution you'd like

Simply z.interpolate_na(['x','y'])

Describe alternatives you've considered

I could extract the data to numpy and interpolate using scipy.interpolate.griddata, but this is not the way xarray should work.

Additional context

No response

What happened:

I am trying to read a HDF file from MODIS satellite product. Source of the file: https://ladsweb.modaps.eosdis.nasa.gov/archive/allData/61/MOD06_L2/2015/012/MOD06_L2.A2015012.1510.061.2017318235218.hdf

and

xr.open_dataset('MOD06_L2.A2015012.1510.061.2017318235218.hdf')

returns

What you expected to happen:

Read the HDF file normally.

Minimal Complete Verifiable Example:

Just

xr.open_dataset('MOD06_L2.A2015012.1510.061.2017318235218.hdf')

Anything else we need to know?:

Environment:

What is your issue?

Hi,

I am trying to access MERRA-2 dataset using opendap links on xarray. The code below is based on a tutorial that @betolink sent me as an example.

The code runs well if parallel=False, but returns OSError: [Errno -70] NetCDF: DAP server error if I set parallel=True, no matter if I create the cluster or not.

@betolink suspected that the workers doesn’t know the authentication and suggested me to do something like mentioned in @rsignell issue.

Which would involve adding client.register_worker_plugin(UploadFile('~/.netrc')) after creating the client. I also tested that but returned the same error. In the code below I had to replace ~/.netrc for the full path because it was returning file not found error.

It is important to say that parallel=True works fine on my local computer using Ubuntu by WSL.

Has anyone faced this problem before or has any guesses on how to solve this issue?

```python

----------------------------------

Import Python modules

----------------------------------

import warnings

warnings.filterwarnings("ignore")

import xarray as xr import matplotlib.pyplot as plt from calendar import monthrange

create_cluster = True parallel = True upload_file = True

if create_cluster: # -------------------------------------- # Creating 50 workers with 1core and 2Gb each # -------------------------------------- import os from dask_jobqueue import SLURMCluster from dask.distributed import Client from dask.distributed import WorkerPlugin

class UploadFile(WorkerPlugin):
    """A WorkerPlugin to upload a local file to workers.
    Parameters
    ----------
    filepath: str
        A path to the file to upload
    Examples
    --------
    >>> client.register_worker_plugin(UploadFile(".env"))
    """
    def __init__(self, filepath):
        """
        Initialize the plugin by reading in the data from the given file.
        """

        self.filename = os.path.basename(filepath)
        self.dirname = os.path.dirname(filepath)
        with open(filepath, "rb") as f:
            self.data = f.read()

    async def setup(self, worker):
        if not os.path.exists(self.dirname):
            os.mkdir(self.dirname)
        os.chdir(self.dirname)
        with open(self.filename, "wb+") as f:
            f.write(self.data)
        return os.listdir()

cluster = SLURMCluster(cores=1, memory="40GB")
cluster.scale(jobs=10)

client = Client(cluster)  # Connect this local process to remote workers
if upload_file:
    client.register_worker_plugin(UploadFile('/home/isimoesdesousa/.netrc'))

---------------------------------

Read data

---------------------------------

MERRA-2 collection (hourly)

collection_shortname = 'M2T1NXAER' collection_longname = 'tavg1_2d_aer_Nx' collection_number = 'MERRA2_400'
MERRA2_version = '5.12.4' year = 2020

Open dataset

Read selected days in the same month and year

month = 1 # January day_beg = 1 day_end = 31

Note that collection_number is MERRA2_401 in a few cases, refer to "Records of MERRA-2 Data Reprocessing and Service Changes"

if year == 2020 and month == 9: collection_number = 'MERRA2_401'

OPeNDAP URL

url = 'https://goldsmr4.gesdisc.eosdis.nasa.gov/opendap/MERRA2/{}.{}/{}/{:0>2d}'.format(collection_shortname, MERRA2_version, year, month) files_month = ['{}/{}.{}.{}{:0>2d}{:0>2d}.nc4'.format(url,collection_number, collection_longname, year, month, days) for days in range(day_beg,day_end+1,1)]

Get the number of files

len_files_month=len(files_month)

Print

print("{} files to be opened:".format(len_files_month)) print("files_month", files_month)

Read dataset URLs

ds = xr.open_mfdataset(files_month, parallel=parallel)

View metadata (function like ncdump -c)

ds ```

As this deals with HPCs, I also posted on pangeo forum https://discourse.pangeo.io/t/access-ges-disc-nasa-dataset-using-xarray-and-dask-on-a-cluster/3195/1

What happened?

python ds.plot.scatter("x","y")

returns:

```python

KeyError Traceback (most recent call last) File /autofs/nas1/home/isimoesdesousa/programs/mambaforge/envs/coringa/lib/python3.9/site-packages/xarray/core/dataset.py:1340, in Dataset._construct_dataarray(self, name) 1339 try: -> 1340 variable = self._variables[name] 1341 except KeyError:

KeyError: None

During handling of the above exception, another exception occurred:

KeyError Traceback (most recent call last) Cell In[60], line 1 ----> 1 ds.plot.scatter("x","y")

File /autofs/nas1/home/isimoesdesousa/programs/mambaforge/envs/coringa/lib/python3.9/site-packages/xarray/plot/accessor.py:1071, in DatasetPlotAccessor.scatter(self, args, kwargs) 1069 @functools.wraps(dataset_plot.scatter) 1070 def scatter(self, args, kwargs) -> PathCollection | FacetGrid[DataArray]: -> 1071 return dataset_plot.scatter(self._ds, *args, kwargs)

File /autofs/nas1/home/isimoesdesousa/programs/mambaforge/envs/coringa/lib/python3.9/site-packages/xarray/plot/dataset_plot.py:914, in scatter(ds, x, y, z, hue, hue_style, markersize, linewidth, figsize, size, aspect, ax, row, col, col_wrap, xincrease, yincrease, add_legend, add_colorbar, add_labels, add_title, subplot_kws, xscale, yscale, xticks, yticks, xlim, ylim, cmap, vmin, vmax, norm, extend, levels, args, kwargs) 912 del locals_["ds"] 913 locals_.update(locals_.pop("kwargs", {})) --> 914 da = temp_dataarray(ds, y, locals) 916 return da.plot.scatter(locals_.pop("args", ()), **locals_)

File /autofs/nas1/home/isimoesdesousa/programs/mambaforge/envs/coringa/lib/python3.9/site-packages/xarray/plot/dataset_plot.py:740, in temp_dataarray(ds, y, locals) 736 coords[key] = ds[key] 738 # The dataarray has to include all the dims. Broadcast to that shape 739 # and add the additional coords: --> 740 _y = ds[y].broadcast_like(ds) 742 return DataArray(_y, coords=coords)

File /autofs/nas1/home/isimoesdesousa/programs/mambaforge/envs/coringa/lib/python3.9/site-packages/xarray/core/dataset.py:1431, in Dataset.getitem(self, key) 1429 return self.isel(**key) 1430 if utils.hashable(key): -> 1431 return self._construct_dataarray(key) 1432 if utils.iterable_of_hashable(key): 1433 return self._copy_listed(key)

File /autofs/nas1/home/isimoesdesousa/programs/mambaforge/envs/coringa/lib/python3.9/site-packages/xarray/core/dataset.py:1342, in Dataset.construct_dataarray(self, name) 1340 variable = self._variables[name] 1341 except KeyError: -> 1342 , name, variable = _get_virtual_variable(self._variables, name, self.dims) 1344 needed_dims = set(variable.dims) 1346 coords: dict[Hashable, Variable] = {}

File /autofs/nas1/home/isimoesdesousa/programs/mambaforge/envs/coringa/lib/python3.9/site-packages/xarray/core/dataset.py:174, in _get_virtual_variable(variables, key, dim_sizes) 171 return key, key, variable 173 if not isinstance(key, str): --> 174 raise KeyError(key) 176 split_key = key.split(".", 1) 177 if len(split_key) != 2:

KeyError: None ```

What did you expect to happen?

To plot the figure:

Minimal Complete Verifiable Example

```Python import pandas as pd

n = 1000 df = pd.DataFrame() df["x"] = np.random.randn(n) df["y"] = np.random.randn(n) ds = df.to_xarray()

this works

ds.plot.scatter(x="x",y="y")

this doesn't work

ds.plot.scatter("x","y") ```

MVCE confirmation

• [x] Minimal example — the example is as focused as reasonably possible to demonstrate the underlying issue in xarray.
• [X] Complete example — the example is self-contained, including all data and the text of any traceback.
• [X] Verifiable example — the example copy & pastes into an IPython prompt or Binder notebook, returning the result.
• [X] New issue — a search of GitHub Issues suggests this is not a duplicate.

Relevant log output

No response

Anything else we need to know?

No response

Environment

Is your feature request related to a problem?

This is not related to any problem, but I think it would be nice to have a support for giving a matplotlib mosaic with the keys for the variables you want to plot for different panels and xarray parse that into the figure.

Describe the solution you'd like

Something like

```python import matplotlib.pyplot as plt import xarray as xr import numpy as np

n = 200 t = np.linspace(0,32np.pi,n) ds = xr.Dataset({letter:(("s","t"),np.sin(t)+0.5*np.random.randn(3,n)) for letter in "A B C D E".split()}) ds = ds.assign_coords(t=t,s=range(3))

mosaic = [ ["A","A","B","B","C","C"], ["X","D","D","E","E","X"], ]

kw = dict(x="t",hue="s",add_legend=False) ds.plot.line(mosaic=mosaic,empty_sentinel="X",**kw)

```

Describe alternatives you've considered

I have a code snippet that generate similar results but with more code.

```python import matplotlib.pyplot as plt import xarray as xr import numpy as np

n = 200 t = np.linspace(0,32np.pi,n) ds = xr.Dataset({letter:(("s","t"),np.sin(t)+0.5*np.random.randn(3,n)) for letter in "A B C D E".split()}) ds = ds.assign_coords(t=t,s=range(3))

mosaic = [ ["A","A","B","B","C","C"], ["X","D","D","E","E","X"], ]

kw = dict(x="t",hue="s",add_legend=False) fig = plt.figure(constrained_layout=True,figsize=(8,4)) ax = fig.subplot_mosaic(mosaic,empty_sentinel="X") for key in ds: ds[key].plot.line(ax=ax[key],**kw) ```

Additional context

No response

Hi,

I am trying to load MUR data from AWS on Google Colab just as Chelle did for her Pangeo tutorial. https://github.com/pangeo-gallery/osm2020tutorial/blob/master/AWS-notebooks/aws_mur_sst_tutorial_long.ipynb

What happened:

```python

ValueError Traceback (most recent call last)

<ipython-input-15-363886c4b27b> in <module>() 1 import xarray as xr 2 ----> 3 ds = xr.open_zarr('https://mur-sst.s3.us-west-2.amazonaws.com/zarr-v1',consolidated=True)

2 frames

/usr/local/lib/python3.7/dist-packages/xarray/backends/plugins.py in get_backend(engine) 133 if engine not in engines: 134 raise ValueError( --> 135 f"unrecognized engine {engine} must be one of: {list(engines)}" 136 ) 137 backend = engines[engine]

ValueError: unrecognized engine zarr must be one of: ['netcdf4', 'scipy', 'store'] ```

What you expected to happen:

Load Zarr data from S3.

Minimal Complete Verifiable Example:

python import xarray as xr ds = xr.open_zarr('https://mur-sst.s3.us-west-2.amazonaws.com/zarr-v1',consolidated=True)

Anything else we need to know?:

I am using Google Colab.

Environment:

Is your feature request related to a problem? Please describe. Everytime I have to do a contourf I need to call it from matplotlib directly because locator argument is not passed from xarray.plot.contourf to matplotlib.

Describe the solution you'd like Being ds a xarray.DataArray, I want to have the behaviour described here when passing locator = ticker.LogLocator() to ds.plot.contourf.

Describe alternatives you've considered I usually have to do plt.contourf(ds.dim_0,ds.dim_1,ds.values,locator = ticker.LogLocator()).