三维交互图不会更新

Question

我正在使用此代码来创建一个交互式图（2d），它的工作原理。

import numpy as np 
import matplotlib.pyplot as plt 
from matplotlib.widgets import Slider, Button, RadioButtons 

fig, ax = plt.subplots() 
plt.subplots_adjust(left=0.25, bottom=0.25) 
t = np.arange(0.0, 1.0, 0.001) 
a0 = 5 
f0 = 3 
s = a0*np.sin(2*np.pi*f0*t) 
l, = plt.plot(t, s, lw=2, color='red') 
plt.axis([0, 1, -10, 10]) 

axcolor = 'lightgoldenrodyellow' 
axfreq = plt.axes([0.25, 0.1, 0.65, 0.03], axisbg=axcolor) 
axamp = plt.axes([0.25, 0.15, 0.65, 0.03], axisbg=axcolor) 

sfreq = Slider(axfreq, 'Freq', 0.1, 30.0, valinit=f0) 
samp = Slider(axamp, 'Amp', 0.1, 10.0, valinit=a0) 


def update(val): 
    amp = samp.val 
    freq = sfreq.val 
    l.set_ydata(amp*np.sin(2*np.pi*freq*t)) 
    fig.canvas.draw_idle() 
sfreq.on_changed(update) 
samp.on_changed(update) 

resetax = plt.axes([0.8, 0.025, 0.1, 0.04]) 
button = Button(resetax, 'Reset', color=axcolor, hovercolor='0.975') 


def reset(event): 
    sfreq.reset() 
    samp.reset() 
button.on_clicked(reset) 

rax = plt.axes([0.025, 0.5, 0.15, 0.15], axisbg=axcolor) 
radio = RadioButtons(rax, ('red', 'blue', 'green'), active=0) 


def colorfunc(label): 
    l.set_color(label) 
    fig.canvas.draw_idle() 
radio.on_clicked(colorfunc) 

plt.show() 

然后我尝试修改它，通过将轴简单地更改为axes3d来创建交互式3d图。我添加了如下所示的导入语句，并将“无花果”和“斧头”的定义替换为下面显示的那些，以变成3d。

from mpl_toolkits.mplot3d import Axes3D 

fig = plt.figure() 
ax = fig.add_subplot(111, projection='3d') 

剧情不再更新，我不明白为什么。似乎函数fig.canvas.draw_idle（）在3d图上不起作用，但我没有其他方式更新图。

任何帮助，将不胜感激，

谢谢！

Answer 1

你可以看到这个函数fig.canvas.draw_idle（）确实存在，并可作为通过观察而变得更新了颜色的预期。

问题在于set_ydata功能，在3D空间中的工作方式不同。 假设你想y坐标更新和z坐标恒定，set_data将给予常量，而set_3d_properties()需要额外的属性被设置为控制y坐标。

这里是工作示例代码：

import numpy as np 
import matplotlib.pyplot as plt 
from mpl_toolkits.mplot3d import Axes3D 
from matplotlib.widgets import Slider, Button, RadioButtons 

fig = plt.figure() 
ax = fig.add_subplot(111, projection="3d") 
plt.subplots_adjust(left=0.25, bottom=0.25) 
t = np.arange(0.0, 1.0, 0.001) 
### create constant z-coordinate 
z = np.zeros_like(t) #    <------------ here 
a0 = 5 
f0 = 3 
s = a0*np.sin(2*np.pi*f0*t) 
l, = plt.plot(t, s, lw=2, color='red') 
plt.axis([0, 1, -10, 10]) 

axfreq = plt.axes([0.25, 0.1, 0.65, 0.03], axisbg="w") 
axamp = plt.axes([0.25, 0.15, 0.65, 0.03], axisbg="w") 

sfreq = Slider(axfreq, 'Freq', 0.1, 30.0, valinit=f0) 
samp = Slider(axamp, 'Amp', 0.1, 10.0, valinit=a0) 


def update(val): 
    amp = samp.val 
    freq = sfreq.val 
    #set constant z coordinate 
    l.set_data(t, z) #    <------------ here 
    # set values to y-coordinate 
    l.set_3d_properties(amp*np.sin(2*np.pi*freq*t), zdir="y") #<------------ here 
    fig.canvas.draw_idle() 

sfreq.on_changed(update) 
samp.on_changed(update) 

resetax = plt.axes([0.8, 0.025, 0.1, 0.04]) 
button = Button(resetax, 'Reset', color=axcolor, hovercolor='0.975') 


def reset(event): 
    sfreq.reset() 
    samp.reset() 
button.on_clicked(reset) 

rax = plt.axes([0.025, 0.5, 0.15, 0.15], axisbg=axcolor) 
radio = RadioButtons(rax, ('red', 'blue', 'green'), active=0) 


def colorfunc(label): 
    l.set_color(label) 
    fig.canvas.draw_idle() 
radio.on_clicked(colorfunc) 

plt.show()