我希望MatPlotLib中的三维散点图在Jupyter Python笔记本中交互旋转。为此,我集成了ipywidgets
中的一个滑块来更新视角。下面的测试代码显示了我要实现的目标。问题是在前一个数字的下面添加了一个新的数字,而不是清除当前的数字。我试过plt.close(fig)
、plt.cla()
和plt.clf()
但没有成功。(进一步地,我意识到在重新创建图形和轴时有开销,但这是我当前关注的较小部分…)
以下是(测试)代码:
# init
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import ipywidgets as widgets
from IPython.display import display
# generate test data
x = np.random.rand(100)
y = np.random.rand(100)
z = np.random.rand(100)
# prepare plot
def draw_plot(angle1 = 20, angle2 = 40):
# create figure
fig = plt.figure(figsize=(15,10))
ax = fig.add_subplot(111, projection='3d')
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')
ax.scatter(x, y, z)
# set view angle
ax.view_init(angle1, angle2)
# show plot
plt.show()
# prepare widgets
angle1_slider = widgets.IntSlider(20, min = 0, max = 60)
angle1_label = widgets.Label(value = 'Angle 1 value is: ' + str(angle1_slider.value))
display(angle1_slider, angle1_label)
# handle angle 1 update
def update_angle1(value):
draw_plot(angle1 = value['new'])
angle1_label.value = 'Angle 1 value is: ' + str(value.new)
angle1_slider.observe(update_angle1, names = 'value')
# draw initial plot
draw_plot()
任何建议都将不胜感激!
你所说的开销是问题的根源。或者换句话说:如果在每次调用函数时,都创建了一个新的图形,那么您最终会得到很多图形是不是很奇怪?
这个想法当然是画一个图形。为了以后能够更新这个图,需要
%matplotlib notebook
后端。更改滑块时调用的函数只需更新视角并重新绘制画布。
这就是它的样子:
相关问题 更多 >
编程相关推荐