diff --git a/skeletor/skeletonize/base.py b/skeletor/skeletonize/base.py index ed7eaa2..548abbb 100644 --- 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