Fix NumPy warnings in some demos.

QuLogic · QuLogic · commit 8d293914ef3f · 2016-12-04T21:13:32.000-05:00
diff --git a/examples/event_handling/lasso_demo.py b/examples/event_handling/lasso_demo.py
@@ -38,9 +38,8 @@ def __init__(self, ax, data):
 
         facecolors = [d.color for d in data]
         self.xys = [(d.x, d.y) for d in data]
-        fig = ax.figure
         self.collection = RegularPolyCollection(
-            fig.dpi, 6, sizes=(100,),
+            6, sizes=(100,),
             facecolors=facecolors,
             offsets=self.xys,
             transOffset=ax.transData)
diff --git a/examples/images_contours_and_fields/streamplot_demo_masking.py b/examples/images_contours_and_fields/streamplot_demo_masking.py
@@ -14,9 +14,9 @@
 speed = np.sqrt(U*U + V*V)
 
 mask = np.zeros(U.shape, dtype=bool)
-mask[40:60, 40:60] = 1
-U = np.ma.array(U, mask=mask)
+mask[40:60, 40:60] = True
 U[:20, :20] = np.nan
+U = np.ma.array(U, mask=mask)
 
 fig, ax = plt.subplots()
 ax.streamplot(X, Y, U, V, color='r')
