Merge pull request #1424 from brian-team/fix_for_matplotlib

Fixes indexing VariableViews with [:, np.newaxis] for subgroups
brian-team · Jun 14, 2022 · 7ed9b6c · 7ed9b6c
2 parents ffe16e8 + bd52ad0
commit 7ed9b6c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/brian2/core/variables.py b/brian2/core/variables.py
@@ -1097,9 +1097,11 @@ def get_with_index_array(self, item):
               len(item) == 2 and
               isinstance(item[0], slice) and
               item[0] == slice(None) and
-              item[1] is None and
-              self.index_var == '_idx'):
-            return variable.get_value()[item]
+              item[1] is None):
+            if self.index_var == '_idx':
+                return variable.get_value()[item]
+            else:
+                return variable.get_value()[self.index_var.get_value()][item]
         else:
             indices = self.indexing(item, self.index_var)