Merge pull request #767 from nkeim/fix-issue-759

soft-matter · Jun 7, 2024 · 09ad08f · 09ad08f
2 parents c1faf16 + 10674d9
commit 09ad08f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/doc/releases/v0.6.txt b/doc/releases/v0.6.txt
@@ -6,6 +6,7 @@ trackpy v0.6.3 is a minor compatibility update for newer Pandas verisons.
 Dependencies
 ~~~~~~~~~~~~
 - Improves support for latest Pandas in emsd function (@hz-xiaxz, #758)
+- Fix deprecated matplotlib usage in 3D plotting (@jacopoabramo, #767)
 
 
 v0.6.2

diff --git a/trackpy/plots.py b/trackpy/plots.py
@@ -80,7 +80,7 @@ def wrapper(*args, **kwargs):
         if kwargs.get('ax') is None:
             if not hasattr(plt.gca(), 'zaxis'):
                 plt.figure()  # initialize new Fig when current axis is not 3d
-            kwargs['ax'] = plt.gca(projection='3d')
+            kwargs['ax'] = plt.subplot(projection='3d')
             show_plot = True
         else:
             if not hasattr(plt.gca(), 'zaxis'):

diff --git a/trackpy/tests/test_plots.py b/trackpy/tests/test_plots.py
@@ -57,6 +57,12 @@ def test_ptraj_t_column(self):
         df.columns = cols
         plots.plot_traj(df, t_column='arbitrary name')
 
+    def test_ptraj3d(self):
+        sparse3d = self.sparse.copy()
+        sparse3d['z'] = 0
+        # smoke test
+        plots.plot_traj3d(sparse3d)
+
     def test_annotate(self):
         suppress_plotting()
         f = DataFrame({'x': [0, 1], 'y': [0, 1], 'frame': [0, 0],