Add display method for unregulargrid

Author: AntSimi

CHANGELOG.rst

@@ -19,6 +19,8 @@ Fixed
 Added
 ^^^^^
 
+- Add display grid for unregular grid
+
 
 [3.6.0] - 2022-01-12
 --------------------

requirements.txt

@@ -1,7 +1,7 @@
 matplotlib
 netCDF4
 numba>=0.53
-numpy<1.21
+numpy<1.22
 opencv-python
 pint
 polygon3

src/py_eddy_tracker/dataset/grid.py

@@ -1074,6 +1074,21 @@ def load(self):
 
             self.init_pos_interpolator()
 
+    def display(self, ax, name, factor=1, **kwargs):
+        """
+        :param matplotlib.axes.Axes ax: matplotlib axes used to draw
+        :param str,array name: variable to display, could be an array
+        :param float factor: multiply grid by
+        :param dict kwargs: look at :py:meth:`matplotlib.axes.Axes.pcolormesh`
+
+        .. minigallery:: py_eddy_tracker.UnRegularGridDataset.display
+        """
+        if "cmap" not in kwargs:
+            kwargs["cmap"] = "coolwarm"
+        data = self.grid(name) if isinstance(name, str) else name
+        print(self.x_c.shape, self.y_c.shape, data.shape)
+        return ax.pcolor(self.x_c, self.y_c, data * factor, **kwargs)
+
     @property
     def bounds(self):
         """Give bounds"""