三維交互圖不會更新

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()