Skeleton: add leafs property

navis-org · Jun 9, 2022 · ce1dea9 · ce1dea9
1 parent 63324a3
commit ce1dea9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/skeletor/skeletonize/base.py b/skeletor/skeletonize/base.py
@@ -118,6 +118,13 @@ def skel_map(self):
                                           ).groupby(0)['index'].apply(np.array
                                                                       ).values
 
+    @property
+    def leafs(self):
+        """Leaf nodes (includes root)."""
+        swc = self.swc
+        leafs = swc[~swc.node_id.isin(swc.parent_id.values) | (swc.parent_id < 0)]
+        return leafs.copy()
+
     def reindex(self, inplace=False):
         """Clean up skeleton."""
         x = self