Merge pull request #1655 from toddrjen/get_segments

pelson · pelson · commit f2ff060d56da · 2013-01-14T01:40:45.000-08:00
add get_segments method to collections.LineCollection
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -1008,6 +1008,16 @@ def set_segments(self, segments):
     set_verts = set_segments  # for compatibility with PolyCollection
     set_paths = set_segments
 
+    def get_segments(self):
+        segments = []
+
+        for path in self._paths:
+            vertices = [vertex for vertex, _ in path.iter_segments()]
+            vertices = np.asarray(vertices)
+            segments.append(vertices)
+
+        return segments
+
     def _add_offsets(self, segs):
         offsets = self._uniform_offsets
         Nsegs = len(segs)
