Make pcolor preserve all data

jklymak · jklymak · commit fdd589e2c881 · 2017-10-30T11:19:48.000-07:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -5273,7 +5273,19 @@ def _pcolorargs(funcname, *args, **kw):
                 raise TypeError('Dimensions of C %s are incompatible with'
                                 ' X (%d) and/or Y (%d); see help(%s)' % (
                                     C.shape, Nx, Ny, funcname))
-            C = C[:Ny - 1, :Nx - 1]
+            X = np.hstack( (1.5 * X[:, [0]] - 0.5 * X[:, [1]],
+                            0.5 * (X[:, 1:] + X[:, :-1]),
+                            1.5 * X[:, [-1]] - 0.5 * X[:, [-2]]) )
+            X = np.vstack( (1.5 * X[[0], :] - 0.5 * X[[1], :],
+                            0.5 * (X[1:, :] + X[:-1, :]),
+                            1.5 * X[[-1], :] - 0.5 * X[[-2], :]) )
+            Y = np.vstack( (1.5 * Y[[0], :] - 0.5 * Y[[1], :],
+                            0.5 * (Y[1:, :] + Y[:-1, :]),
+                            1.5 * Y[[-1], :] - 0.5 * Y[[-2], :]) )
+            Y = np.hstack( (1.5 * Y[:, [0]] - 0.5 * Y[:, [1]],
+                            0.5 * (Y[:, 1:] + Y[:, :-1]),
+                            1.5 * Y[:, [-1]] - 0.5 * Y[:, [-2]]) )
+
         C = cbook.safe_masked_invalid(C)
         return X, Y, C
 
