ghiggi / gpm_api

Global Precipitation Measurement Mission (GPM) python package to download and analyze data with xarray
https://gpm-api.readthedocs.io
MIT License
46 stars 4 forks source link

[BUG] plot_map strange behavior with NaN coordinates #41

Closed sphamba closed 4 months ago

sphamba commented 4 months ago

Is there an existing issue for this?

Current Behavior

Distorded figure

Expected Behavior

No response

Steps To Reproduce

import matplotlib.pyplot as plt
import numpy as np
import xarray as xr
from gpm_api.visualization import plot

n_cross_track = 5
n_along_track = 10

cross_track = np.arange(n_cross_track)
along_track = np.arange(n_along_track)
data = np.linspace(0, 1, n_along_track * n_cross_track).reshape(n_cross_track, n_along_track)
granule_id = np.zeros(n_along_track)

lon = np.linspace(0, 40, n_along_track)
lat = np.linspace(0, 20, n_cross_track)
lon, lat = np.meshgrid(lon, lat)

# Introduce NaN coordinates
lon[1, :] = np.nan

da = xr.DataArray(data, coords={"cross_track": cross_track, "along_track": along_track})
da.coords["lat"] = (("cross_track", "along_track"), lat)
da.coords["lon"] = (("cross_track", "along_track"), lon)
da.coords["gpm_granule_id"] = ("along_track", granule_id)

plot.plot_map(da)
plt.show()

Environment

- Python: 3.9.18

Anything else?

No response

ghiggi commented 4 months ago

Fixed by https://github.com/ghiggi/gpm_api/pull/44