使用matplotlib更改3D散点图的默认颜色条

Question

我有3D坐标定义了我的3D散点图中点的位置,并且我想要创建一个颜色标度来表示相应的值范围,这样我可以创建相应数量的值:

from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

xs,ys,zs = np.random.random(50),np.random.random(50),np.random.random(50)
values = np.random.random(50)*10

p = ax.scatter3D(xs, ys, zs=zs, c=values)
cbar = fig.colorbar(p)
cbar.cmap(plt.cm.Greys)

没有最后一行cbar.cmap(plt.cm.Greys)我得到一个默认的colormap(喷射我认为),但是当我尝试修改它时,我得到了错误TypeError: Cannot cast array data from dtype('O') to dtype('int64') according to the rule 'safe'.我希望能够改变这一点,我一直在查看文档,但似乎有很多不同的方法来做类似的事情.如何从这一点更改默认色图？

我

Answer 1

你必须给出cmap分散的关键字.以下是所有可能的cmaps:http: //matplotlib.org/examples/color/colormaps_reference.html ,您可能希望将ax添加到颜色栏.

from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

xs,ys,zs = np.random.random(50),np.random.random(50),np.random.random(50)
values = np.random.random(50)*10

p = ax.scatter3D(xs, ys, zs=zs, c=values, cmap='hot')

fig.colorbar(p, ax=ax)