Merge pull request #15525 from meeseeksmachine/auto-backport-of-pr-15498-on-v3.2.x

timhoffm · web-flow · commit a22443fe358a · 2019-10-26T09:52:59.000+02:00
Backport PR #15498 on branch v3.2.x (Simplify matshow example)
diff --git a/examples/images_contours_and_fields/matshow.py b/examples/images_contours_and_fields/matshow.py
@@ -3,22 +3,15 @@
 Matshow
 =======
 
-Simple `~.axes.Axes.matshow` example.
+`~.axes.Axes.matshow` visualizes a 2D matrix or array as color-coded image.
 """
 import matplotlib.pyplot as plt
 import numpy as np
 
+# a 2D array with linearly increasing values on the diagonal
+a = np.diag(range(15))
 
-def samplemat(dims):
-    """Make a matrix with all zeros and increasing elements on the diagonal"""
-    aa = np.zeros(dims)
-    for i in range(min(dims)):
-        aa[i, i] = i
-    return aa
-
-
-# Display matrix
-plt.matshow(samplemat((15, 15)))
+plt.matshow(a)
 
 plt.show()
 
