Move axis around to get right ordering

yt-project · Feb 22, 2019 · ebcf82a · ebcf82a
1 parent 937b10f
commit ebcf82a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/yt/geometry/oct_container.pyx b/yt/geometry/oct_container.pyx
@@ -665,9 +665,12 @@ cdef class OctreeContainer:
         output_data = {}
         for key, val in input_data.items():
             val = val.T.reshape(-1)
-            output_data[key] = subset.apply_units(
-                np.where(icell_inds>=0, val[icell_inds], np.nan),
-                val.units)
+
+            output_data[key] = np.moveaxis(
+                subset.apply_units(
+                    np.where(icell_inds>=0, val[icell_inds], np.nan),
+                    val.units), 0, -1)
+
         return output_data