Merge pull request #19172 from ufmayer/master

jklymak · web-flow · commit 28efb116d819 · 2020-12-25T12:15:49.000-08:00
Fix 3D surface example bug for non-square grid
diff --git a/examples/mplot3d/surface3d_3.py b/examples/mplot3d/surface3d_3.py
@@ -28,7 +28,7 @@
 colors = np.empty(X.shape, dtype=str)
 for y in range(ylen):
     for x in range(xlen):
-        colors[x, y] = colortuple[(x + y) % len(colortuple)]
+        colors[y, x] = colortuple[(x + y) % len(colortuple)]
 
 # Plot the surface with face colors taken from the array we made.
 surf = ax.plot_surface(X, Y, Z, facecolors=colors, linewidth=0)
