Fixed N+1/M+1 case

jklymak · jklymak · commit 451fee263ef9 · 2017-10-30T12:20:27.000-07:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -5273,18 +5273,20 @@ 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))
-            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], :],
+            if numCols == Nx:
+                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]]) )
+                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]]) )
+            if numRows == Ny:
+                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], :],
+                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
