使用python中的滑块更改散点图中点的大小和颜色

2024-09-28 17:19:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图使用一个滑块来改变三维散点图中几个点的大小和颜色。我很困惑为什么颜色不会更新,而且我还没有找到一个命令来更新尺寸。在

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

Tags: fromimporttimematplotlibonfigcolax
1条回答
网友
1楼 · 发布于 2024-09-28 17:19:53

当然不是最优雅也最正确的方法来完成您所要求的,但是在您弄清楚之前,您可以在每次调用函数时重新初始化axes3dsublot。在这种情况下,如下所示。在

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

相关问题 更多 >