add numpy refactror for errorbar3d example

vlas-sokolov · eric-wieser · web-flow · commit 3a99be8630f3 · 2020-05-20T20:11:51.000+02:00
Co-authored-by: Eric Wieser &lt;wieser.eric@gmail.com&gt;
diff --git a/examples/mplot3d/errorbar3d.py b/examples/mplot3d/errorbar3d.py
@@ -17,10 +17,9 @@
 x, y, z = np.sin(t), np.cos(3*t), np.sin(5*t)
 
 estep = 15
-zuplims = [True if (not i % estep and i // estep % 3 == 0)
-           else False for i in range(t.size)]
-zlolims = [True if (not i % estep and i // estep % 3 == 2)
-           else False for i in range(t.size)]
+i = np.arange(t.size)
+zuplims = (i % estep == 0) & (i // estep % 3 == 0)
+zlolims = (i % estep == 0) & (i // estep % 3 == 2)
 
 ax.errorbar(x, y, z, 0.2, zuplims=zuplims, zlolims=zlolims, errorevery=estep)
 
