Fix lasso_demo.py

mdboom · mdboom · commit f0231dce640f · 2008-01-18T17:27:35.000Z
svn path=/trunk/matplotlib/; revision=4877
diff --git a/examples/lasso_demo.py b/examples/lasso_demo.py
@@ -35,12 +35,12 @@ def __init__(self, ax, data):
 
         self.Nxy = len(data)
 
-        self.facecolors = [d.color for d in data]
+        facecolors = [d.color for d in data]
         self.xys = [(d.x, d.y) for d in data]
 
         self.collection = RegularPolyCollection(
             fig.dpi, 6, sizes=(100,),
-            facecolors=self.facecolors,
+            facecolors=facecolors,
             offsets = self.xys,
             transOffset = ax.transData)
 
@@ -49,12 +49,13 @@ def __init__(self, ax, data):
         self.cid = self.canvas.mpl_connect('button_press_event', self.onpress)
 
     def callback(self, verts):
+        facecolors = self.collection.get_facecolors()
         ind = nonzero(points_inside_poly(self.xys, verts))[0]
         for i in range(self.Nxy):
             if i in ind:
-                self.facecolors[i] = Datum.colorin
+                facecolors[i] = Datum.colorin
             else:
-                self.facecolors[i] = Datum.colorout
+                facecolors[i] = Datum.colorout
 
         self.canvas.draw_idle()
         self.canvas.widgetlock.release(self.lasso)
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -275,6 +275,10 @@ def set_facecolor(self, c):
 
     set_facecolors = set_facecolor
 
+    def get_facecolor(self):
+        return self._facecolors
+    get_facecolors = get_facecolor
+
     def set_edgecolor(self, c):
         """
         Set the edgecolor(s) of the collection. c can be a matplotlib color
