首先说明我所参考的资料:
Matplotlib:给子图添加colorbar(颜色条或渐变色条) - 简书 (jianshu.com)
使用Matplotlib.pyplot绘制三维图的方法很简单:
#创建子图ax,并且将他的参数projection设置为3d即可
ax=plt.subplot(projection='3d')
但接下来实现颜色映射就有问题了,以下代码中的ratio是我dataframe数据中的一列。
#绘制散点图,并设置它的颜色参数c
ax.scatter(x,y,z,c=ratio)
绘制的图长这个样子:
为了能够绘制出颜色条,以下是我的操作:
fig=plt.figure()
ax0=ax.scatter(x,y,z,c=ratio)
fig.colorbar(ax0)
最终的效果: