12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix matplotlib deprecation

release/4.3a0
Varun Agrawal 2021-09-03 08:05:11 -04:00
parent 67a26c1f93
commit 5131f6b0a6
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
python/gtsam/utils/plot.py
View File

@ -23,7 +23,10 @@ def set_axes_equal(fignum: int) -> None:
fignum: An integer representing the figure number for Matplotlib. fignum: An integer representing the figure number for Matplotlib.
""" """
fig = plt.figure(fignum) fig = plt.figure(fignum)
ax = fig.gca(projection='3d') if not fig.axes:
ax = fig.add_subplot(projection='3d')
else:
ax = fig.axes[0]
limits = np.array([ limits = np.array([
ax.get_xlim3d(), ax.get_xlim3d(),
@ -339,7 +342,11 @@ def plot_pose3(
""" """
# get figure object # get figure object
fig = plt.figure(fignum) fig = plt.figure(fignum)
axes = fig.gca(projection='3d') if not fig.axes:
axes = fig.add_subplot(projection='3d')
else:
axes = fig.axes[0]
plot_pose3_on_axes(axes, pose, P=P, axis_length=axis_length) plot_pose3_on_axes(axes, pose, P=P, axis_length=axis_length)
axes.set_xlabel(axis_labels[0]) axes.set_xlabel(axis_labels[0])