Simplify curvilinear grid examples.

anntzer · anntzer · commit 20ac9ef903d8 · 2021-07-17T18:35:11.000+02:00
Since 8840d6e we ensure that `x` and `y` passed to `tr` and `inv_tr` are already arrays (see the definition of `_UserTransform2D.transform_non_affine`), so we don't need additional casting-to-arrays.
diff --git a/examples/axisartist/demo_curvelinear_grid.py b/examples/axisartist/demo_curvelinear_grid.py
@@ -27,24 +27,17 @@ def curvelinear_test1(fig):
     Grid for custom transform.
     """
 
-    def tr(x, y):
-        x, y = np.asarray(x), np.asarray(y)
-        return x, y - x
-
-    def inv_tr(x, y):
-        x, y = np.asarray(x), np.asarray(y)
-        return x, y + x
+    def tr(x, y): return x, y - x
+    def inv_tr(x, y): return x, y + x
 
     grid_helper = GridHelperCurveLinear((tr, inv_tr))
 
     ax1 = fig.add_subplot(1, 2, 1, axes_class=Axes, grid_helper=grid_helper)
-    # ax1 will have a ticks and gridlines defined by the given
-    # transform (+ transData of the Axes). Note that the transform of
-    # the Axes itself (i.e., transData) is not affected by the given
-    # transform.
-
-    xx, yy = tr([3, 6], [5, 10])
-    ax1.plot(xx, yy, linewidth=2.0)
+    # ax1 will have ticks and gridlines defined by the given transform (+
+    # transData of the Axes).  Note that the transform of the Axes itself
+    # (i.e., transData) is not affected by the given transform.
+    xx, yy = tr(np.array([3, 6]), np.array([5, 10]))
+    ax1.plot(xx, yy)
 
     ax1.set_aspect(1)
     ax1.set_xlim(0, 10)
diff --git a/examples/axisartist/demo_curvelinear_grid2.py b/examples/axisartist/demo_curvelinear_grid2.py
@@ -24,14 +24,10 @@ def curvelinear_test1(fig):
     """Grid for custom transform."""
 
     def tr(x, y):
-        sgn = np.sign(x)
-        x, y = np.abs(np.asarray(x)), np.asarray(y)
-        return sgn*x**.5, y
+        return np.sign(x)*abs(x)**.5, y
 
     def inv_tr(x, y):
-        sgn = np.sign(x)
-        x, y = np.asarray(x), np.asarray(y)
-        return sgn*x**2, y
+        return np.sign(x)*x**2, y
 
     grid_helper = GridHelperCurveLinear(
         (tr, inv_tr),
