Merge pull request #15498 from timhoffm/doc-matshow

dstansby · web-flow · commit 8be73a3ecb4d · 2019-10-26T00:06:28.000+01:00
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()
 
