FIX: make pcolormesh sample unevenly

jklymak · jklymak · commit 42df67e2b195 · 2021-04-01T11:27:05.000-07:00
diff --git a/lib/matplotlib/mpl-data/stylelib/cheatsheet_gallery.mplstyle b/lib/matplotlib/mpl-data/stylelib/cheatsheet_gallery.mplstyle
@@ -13,4 +13,4 @@ figure.subplot.top: 0.99
 
 # colors:
 image.cmap    : Oranges
-axes.prop_cycle : cycler('color', ['ff7f0e', '1f77b4', '2ca02c'])
+axes.prop_cycle: cycler('color', ['FF7F0E', '1F77B4', '2CA02C'])
diff --git a/plot_types/A_basic/f_pie.py b/plot_types/A_basic/f_pie.py
@@ -7,11 +7,10 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-
 # plot
 with plt.style.context('cheatsheet_gallery'):
     fig, ax = plt.subplots()
-    
+
     # make data
     X = [1, 2, 3, 4]
     colors = np.zeros((len(X), 4))
@@ -28,4 +27,4 @@
     ax.set_ylim(0, 8)
     ax.set_yticks(np.arange(1, 8))
 
-plt.show()
+    plt.show()
diff --git a/plot_types/B_arrays/b_pcolormesh.py b/plot_types/B_arrays/b_pcolormesh.py
@@ -10,18 +10,23 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-# make data
+# make full-res data
 X, Y = np.meshgrid(np.linspace(-3, 3, 256), np.linspace(-3, 3, 256))
 Z = (1 - X/2. + X**5 + Y**3) * np.exp(-X**2 - Y**2)
 Z = Z - Z.min()
 
+# sample unevenly in x:
+dx = np.sqrt((np.arange(16) - 8)**2) + 6
+dx = np.floor(dx / sum(dx) * 255)
+xint = np.cumsum(dx).astype('int')
+X = X[0, xint]
+Y = Y[::8, 0]
+Z = Z[::8, :][:, xint]
+
 # plot
 with plt.style.context('cheatsheet_gallery'):
     fig, ax = plt.subplots()
 
     # plot:
-    ax.pcolormesh(X, Y, Z, vmin=0, vmax=1.1)
-
-    #ax.set_ylim(np.min(Y), np.max(Y))
-    #ax.set_xlim(np.min(X), np.max(X))
+    ax.pcolormesh(X, Y, Z, vmin=0, vmax=1.5)
     plt.show()
diff --git a/plot_types/B_arrays/d_quiver.py b/plot_types/B_arrays/d_quiver.py
@@ -23,4 +23,4 @@
     ax.set_ylim(0, 8)
     ax.set_yticks(np.arange(1, 8))
 
-plt.show()
+    plt.show()
diff --git a/plot_types/B_arrays/e_streamplot.py b/plot_types/B_arrays/e_streamplot.py
@@ -22,4 +22,4 @@
     # plot stream plot
     ax.streamplot(X[1:, 1:], Y[1:, 1:], U, V, zorder=2)
 
-plt.show()
+    plt.show()
diff --git a/plot_types/C_stats/a_hist.py b/plot_types/C_stats/a_hist.py
@@ -21,4 +21,4 @@
     ax.set_ylim(0, 80)
     ax.set_yticks(np.arange(1, 80, 10))
 
-plt.show()
+    plt.show()
diff --git a/plot_types/C_stats/b_boxplot.py b/plot_types/C_stats/b_boxplot.py
@@ -13,22 +13,21 @@
 # plot
 with plt.style.context('cheatsheet_gallery'):
     fig, ax = plt.subplots()
-    orange = '#ff7f0e'
     VP = ax.boxplot(D, positions=[2, 4, 6], widths=1.5, patch_artist=True,
                     showmeans=False, showfliers=False,
                     medianprops={"color": "white",
                                  "linewidth": 0.5},
-                    boxprops={"facecolor": orange,
+                    boxprops={"facecolor": "C0",
                               "edgecolor": "white",
                               "linewidth": 0.5},
-                    whiskerprops={"color": orange,
+                    whiskerprops={"color": "C0",
                                   "linewidth": 1.5},
-                    capprops={"color": orange,
+                    capprops={"color": "C0",
                               "linewidth": 1.5})
 
     ax.set_xlim(0, 8)
     ax.set_xticks(np.arange(1, 8))
     ax.set_ylim(0, 8)
     ax.set_yticks(np.arange(1, 8))
 
-plt.show()
+    plt.show()
diff --git a/plot_types/C_stats/c_errorbar.py b/plot_types/C_stats/c_errorbar.py
@@ -16,11 +16,11 @@
 with plt.style.context('cheatsheet_gallery'):
     fig, ax = plt.subplots()
 
-    ax.errorbar(X, Y, E, color="C1", linewidth=2, capsize=6)
+    ax.errorbar(X, Y, E, linewidth=2, capsize=6)
 
     ax.set_xlim(0, 8)
     ax.set_xticks(np.arange(1, 8))
     ax.set_ylim(0, 8)
     ax.set_yticks(np.arange(1, 8))
 
-plt.show()
+    plt.show()
diff --git a/plot_types/C_stats/d_violin.py b/plot_types/C_stats/d_violin.py
@@ -18,11 +18,12 @@
                        showmeans=False, showmedians=False, showextrema=False)
     #style:
     for body in VP['bodies']:
+
         body.set_alpha(0.9)
 
-ax.set_xlim(0, 8)
-ax.set_xticks(np.arange(1, 8))
-ax.set_ylim(0, 8)
-ax.set_yticks(np.arange(1, 8))
+    ax.set_xlim(0, 8)
+    ax.set_xticks(np.arange(1, 8))
+    ax.set_ylim(0, 8)
+    ax.set_yticks(np.arange(1, 8))
 
-plt.show()
+    plt.show()
diff --git a/plot_types/C_stats/f_eventplot.py b/plot_types/C_stats/f_eventplot.py
@@ -15,12 +15,11 @@
 with plt.style.context('cheatsheet_gallery'):
     fig, ax = plt.subplots()
 
-    ax.eventplot(D, colors="C0", orientation="vertical",
-                 lineoffsets=X, linewidth=0.75)
+    ax.eventplot(D, orientation="vertical", lineoffsets=X, linewidth=0.75)
 
     ax.set_xlim(0, 8)
     ax.set_xticks(np.arange(1, 8))
     ax.set_ylim(0, 8)
     ax.set_yticks(np.arange(1, 8))
 
-plt.show()
+    plt.show()
diff --git a/plot_types/C_stats/g_hist2d.py b/plot_types/C_stats/g_hist2d.py
@@ -17,7 +17,7 @@
 
     ax.hist2d(x, y, bins=(np.arange(-3, 3, 0.1), np.arange(-3, 3, 0.1)))
 
-ax.set_xlim(-2, 2)
-ax.set_ylim(-3, 3)
+    ax.set_xlim(-2, 2)
+    ax.set_ylim(-3, 3)
 
-plt.show()
+    plt.show()
diff --git a/plot_types/C_stats/h_hexbin.py b/plot_types/C_stats/h_hexbin.py
@@ -17,7 +17,7 @@
 
     ax.hexbin(x, y, gridsize=20)
 
-ax.set_xlim(-2, 2)
-ax.set_ylim(-3, 3)
+    ax.set_xlim(-2, 2)
+    ax.set_ylim(-3, 3)
 
-plt.show()
+    plt.show()
