FIX: make pcolormesh sample unevenly

jklymak · jklymak · commit 72f35a4bb10f · 2021-03-16T14:05:57.000-07:00
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()
