From ce1dea9ebb40b325e31af63d2fad8a64f4154042 Mon Sep 17 00:00:00 2001
From: Philipp Schlegel <observing@web.de>
Date: Thu, 9 Jun 2022 10:54:42 +0100
Subject: [PATCH] Skeleton: add `leafs` property

---
 skeletor/skeletonize/base.py | 7 +++++++
 1 file changed, 7 insertions(+)

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