python - 将颜色贴图应用于 mpl_toolkits.mplot3d.Axes3D.bar3d

Question

Axes3D 的 bar3d 函数有一个“颜色”参数，它可以接受数组来为单个条形着色不同的颜色 - 但是我将如何以与 plot_surface 函数相同的方式应用颜色图(即 cmap = cm.jet)？这将使特定高度的条形成为反射(reflect)其高度的颜色。

http://matplotlib.sourceforge.net/examples/mplot3d/hist3d_demo.html

http://matplotlib.sourceforge.net/mpl_toolkits/mplot3d/api.html

Answer 1

按照 Ferguzz 提供的答案，这里有一个更完整/最新的解决方案:

import matplotlib.colors as colors
import matplotlib.cm as cm


dz = height_values
offset = dz + np.abs(dz.min())
fracs = offset.astype(float)/offset.max()
norm = colors.Normalize(fracs.min(), fracs.max())
color_values = cm.jet(norm(fracs.tolist()))
ax.bar3d(xpos,ypos,zpos,1,1,dz, color=color_values)

请注意以下几点:

您应该定义与 https://matplotlib.org/examples/pylab_examples/hist_colormapped.html 中的代码类似的所有变量(例如 xpos、ypos)。

normalize() 现在是 Normalize()

fracs 的类型为 Series(来自 Pandas )，必须转换为列表