NeuronList: fix issue with adding IDs to combined dataframes

navis-org · Aug 24, 2020 · 2e3d55f · 2e3d55f
1 parent c4ad785
commit 2e3d55f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/navis/core/neuronlist.py b/navis/core/neuronlist.py
@@ -277,12 +277,14 @@ def __getattr__(self, key):
                                ignore_index=True,
                                join='outer',
                                sort=True)
+
+                # For each row label which neuron (id) it belongs to
                 df['neuron'] = None
                 ix = 0
                 for k, v in enumerate(values):
                     if isinstance(v, pd.DataFrame):
-                        df.iloc[ix:ix:v.shape[0],
-                                df.columns.get_loc('neuron')] = k
+                        df.iloc[ix:v.shape[0],
+                                df.columns.get_loc('neuron')] = self.neurons[k].id
                         ix += v.shape[0]
                 return df
             elif all(is_quantity):