从Matplotlib文档页面上的示例中借用并稍作修改,
import numpy as np from mpl_toolkits.mplot3d import Axes3D import matplotlib.pyplot as plt def randrange(n, vmin, vmax): return (vmax-vmin)*np.random.rand(n) + vmin fig = plt.figure() ax = fig.add_subplot(111, projection='3d') n = 100 for c, m, zl, zh in [('r', 'o', -50, -25), ('b', '^', -30, -5)]: xs = randrange(n, 23, 32) ys = randrange(n, 0, 100) zs = randrange(n, zl, zh) cs = randrange(n, 0, 100) ax.scatter(xs, ys, zs, c=cs, marker=m) ax.set_xlabel('X Label') ax.set_ylabel('Y Label') ax.set_zlabel('Z Label') plt.show()
将给出每个点具有不同颜色的3D散点图(在此示例中为随机颜色)。向图中添加颜色条的正确方法是什么,因为添加plt.colorbar()或ax.colorbar()似乎不起作用。
plt.colorbar()
ax.colorbar()
这会产生一个颜色条(尽管可能不是您需要的颜色条):
替换此行:
ax.scatter(xs, ys, zs, c=cs, marker=m)
与
p = ax.scatter(xs, ys, zs, c=cs, marker=m)
然后使用
fig.colorbar(p)
接近尾声