我有4个要绘制的变量,我有一个字典,其中包含我需要的所有数据,这些数据给出了x,y,z和c数据。我想使用x,y和z的位置绘制每个数据点的位置,并使用从0到1的范围对第四个变量进行着色。绘制,但不知道如何向其中添加颜色条。
这是我的代码中用于绘图的部分:
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.set_xlabel('X (m)')
ax.set_ylabel('Y (m)')
ax.set_zlabel('Z (m)')
ax.set_xlim(0,0.1)
ax.set_ylim(0,0.04)
ax.set_zlim(-0.02,0.04)
cmp = plt.cm.get_cmap('RdYlBu')
for i in range(1,len(coordinates)):
x = coordinates[i,0]
y = coordinates[i,1]
X = plot_data["x-" + str(x) + "-y" + str(y)][:,0]
Y = plot_data["x-" + str(x) + "-y" + str(y)][:,1]
Z = plot_data["x-" + str(x) + "-y" + str(y)][:,2]
c = plot_data["x-" + str(x) + "-y" + str(y)][:,3]
ax.scatter(X,Y,Z,c=c,vmin=0,vmax=1,cmap=cmp)
fig.colorbar(ax)
plt.show()
代码在我的字典中正常循环,并在删除时产生我满意的情节:
fig.colorbar(ax)
当我将其包括在内时,情节就崩溃了,有没有办法为所有正在分别作图的数据实现颜色条?