Replace loop with array operations in an example.

QuLogic · QuLogic · commit d194bc861d16 · 2020-02-15T01:57:03.000-05:00
diff --git a/examples/scales/logit_demo.py b/examples/scales/logit_demo.py
@@ -14,8 +14,7 @@
 xmax = 10
 x = np.linspace(-xmax, xmax, 10000)
 cdf_norm = [math.erf(w / np.sqrt(2)) / 2 + 1 / 2 for w in x]
-cdf_laplacian = [1 / 2 * np.exp(w) if w < 0 else 1 - 1 / 2 * np.exp(-w)
-                 for w in x]
+cdf_laplacian = np.where(x < 0, 1 / 2 * np.exp(x), 1 - 1 / 2 * np.exp(-x))
 cdf_cauchy = np.arctan(x) / np.pi + 1 / 2
 
 fig, axs = plt.subplots(nrows=3, ncols=2, figsize=(6.4, 8.5))
