From 291a54cd5444708d6b4d489c2601ce10eed329eb Mon Sep 17 00:00:00 2001
From: Philipp Schlegel <observing@web.de>
Date: Mon, 27 Jan 2025 10:16:42 +0000
Subject: [PATCH] NeuronList: add `masked` property to summary

---
 navis/core/neuronlist.py | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/navis/core/neuronlist.py b/navis/core/neuronlist.py
index ab5e6a61..a59f4f16 100644
--- a/navis/core/neuronlist.py
+++ b/navis/core/neuronlist.py
@@ -780,12 +780,23 @@ def summary(self,
         if not isinstance(N, slice):
             N = slice(N)
 
-        return pd.DataFrame(data=[[getattr(n, a, 'NA') for a in props]
-                                  for n in config.tqdm(self.neurons[N],
-                                                       desc='Summarizing',
-                                                       leave=False,
-                                                       disable=not progress)],
-                            columns=props)
+        summary = pd.DataFrame(
+            data=[
+                [getattr(n, a, "NA") for a in props]
+                for n in config.tqdm(
+                    self.neurons[N],
+                    desc="Summarizing",
+                    leave=False,
+                    disable=not progress,
+                )
+            ],
+            columns=props,
+        )
+
+        if any((n.is_masked for n in self.neurons[N])):
+            summary['masked'] = [n.is_masked for n in self.neurons[N]]
+
+        return summary
 
     def itertuples(self):
         """Helper to mimic `pandas.DataFrame.itertuples()`."""