Add 'get_segments' method to the collections.LineCollection class

toddrjen · toddrjen · commit faf58d54d7af · 2013-01-14T10:35:00.000+01:00
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)
