使用滑块在Python中改变散点图中点的大小和颜色

Question

我正在尝试使用一个滑块来改变3D散点图中几个点的大小和颜色。可是我搞不懂为什么颜色没有更新，而且我也找不到可以更新大小的命令。

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import pyplot
from matplotlib import pyplot as plt
from matplotlib.widgets import Slider

def on_change(val):
    time=int(val)/1
    point1.set_color(col[time][0])
    point2.set_color(col[time][1])
    #point1.size(sizes[time][0])
    #point2.size(sizes[time][1])
    fig.canvas.draw()

x=[[.3,.7],[.3,.7],[.3,.7]]
y=[[0.5,0.5],[0.5,0.5],[0.5,0.5]]
z=[[0.5,0.5],[0.5,0.5],[0.5,0.5]]
p=0
col=[['blue','red'],['green','green'],['red','blue']]
sizes=[[100,10],[55,55],[10,100]]
fig = pyplot.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot([0,0,0,0,0,1,1,0,0,1,1,1,1,1,1,0],[0,1,1,0,0,0,0,0,1,1,0,0,1,1,1,1],[0,0,1,1,0,0,1,1,1,1,1,0,0,1,0,0],c='black',zorder=10)
point1=ax.scatter(x[p][0],y[p][0],z[p][0],s=sizes[p][0],color=col[p][0],zorder=0)
point2=ax.scatter(x[p][1],y[p][1],z[p][1],s=sizes[p][1],color=col[p][1],zorder=0)
ax.set_xlabel('Width')
ax.set_ylabel('Depth')
ax.set_zlabel('Height')
slider_ax = plt.axes([0.15, 0.05, 0.7, 0.02])
slider = Slider(slider_ax, "min", 0, 2, valinit=1, color='blue')
slider.on_changed(on_change)
pyplot.show()

Answer 1

当然，这不是最优雅或最正确的方法来完成你所要求的事情，但在你弄清楚之前，你可以在每次调用你的函数时重新初始化你的 Axes3DSubplot。下面是具体的做法。

def on_change(val):
   time=int(val)/1   
   global ax
   ax = fig.add_subplot(111, projection='3d')
   ax.plot([0,0,0,0,0,1,1,0,0,1,1,1,1,1,1,0],[0,1,1,0,0,0,0,0,1,1,0,0,1,1,1,1],[0,0,1,1,0,0,1,1,1,1,1,0,0,1,0,0],c='black',zorder=10)

   p = 0
   ax.scatter(x[p][0],y[p][0],z[p][0],s=sizes[time][0],color=col[time][0],zorder=0)
   ax.scatter(x[p][1],y[p][1],z[p][1],s=sizes[time][1],color=col[time][1],zorder=0)

   fig.canvas.draw()