在python中以交互方式旋转3D绘图 - matplotlib

时间:2018-04-23 12:38:54

标签: python python-3.x matplotlib 3d interactive

我想知道如何以this视频中描述的方式交互式旋转3D绘图(如果您从上方或下方或从右侧或左侧决定)。我可以在spyder或jupyter Notebook中生成3D绘图,但之后它仍然是静态的,我无法与它交互并旋转/改变视点的角度。

以下是代码:

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')

scale = 8
# Make data.
X = np.arange(-scale, scale, 0.25)
Y = np.arange(-scale, scale, 0.25)
X, Y = np.meshgrid(X, Y)
Z = X**2 + Y**2

# Plot the surface.
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
                   linewidth=0, antialiased=False)

# Customize the z axis.
ax.set_zlim(0, 100)
ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

# rotate the axes and update
for angle in range(0, 360):
   ax.view_init(30, 40)

# Add a color bar which maps values to colors.
fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

非常感谢你!

1 个答案:

答案 0 :(得分:5)

好的,我在另一篇文章中找到了答案:

How can I open the interactive Matplotlib window in IPython notebook?

必须在脚本开头写下以下行%matplotlib qt +你必须重新启动jupyter笔记本才能正常工作

非常感谢你的帮助!