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],