NeuronList indexer: fix issue when key is pd.Series with non-range index

navis-org · Mar 4, 2025 · fe544a3 · fe544a3
1 parent b2d8c7b
commit fe544a3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/navis/core/neuronlist.py b/navis/core/neuronlist.py
@@ -422,6 +422,10 @@ def __deepcopy__(self):
 
     def __getitem__(self, key):
         if utils.is_iterable(key):
+            # This avoids issues when the Series' index is not [0, 1, 2, etc]
+            if isinstance(key, pd.Series):
+                key = key.values
+
             if all([isinstance(k, (bool, np.bool_)) for k in key]):
                 if len(key) != len(self.neurons):
                     raise IndexError('boolean index did not match indexed '