Merge pull request #101 from traja-team/fix/trip-grid

Fix histogram rotation to keep y as first dimension
traja-team · Oct 11, 2022 · d4c5ff3 · d4c5ff3
2 parents f347c8c + 5968833
commit d4c5ff3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/traja/__init__.py b/traja/__init__.py
@@ -9,7 +9,7 @@
 from .trajectory import *
 
 __author__ = "justinshenk"
-__version__ = "0.2.13"
+__version__ = "22.0.0"
 
 logging.basicConfig(level=logging.INFO)
 

diff --git a/traja/plotting.py b/traja/plotting.py
@@ -966,11 +966,14 @@ def trip_grid(
     ymin, ymax = ylim
 
     x, y = zip(*df.values)
-    # FIXME: Remove redundant histogram calculation
+
     hist, x_edges, y_edges = np.histogram2d(
         x, y, bins, range=((xmin, xmax), (ymin, ymax)), normed=normalize
     )
 
+    # rotate to keep y as first dimension
+    hist = np.rot90(hist)
+
     if log:
         hist = np.log(hist + np.e)
     if hist_only:  # TODO: Evaluate potential use cases or remove