aplique o mapa de cores para mpl_toolkits.mplot3d.Axes3D.bar3d

Existe um argumento 'color' para a função bar3d do Axes3D que pode aceitar arrays para colorir barras individuais de cores diferentes - mas como eu aplicaria um mapa de cores (ex: cmap = cm.jet) da mesma forma que uma função plot_surface por exemplo? Isso tornaria uma barra de certa altura uma cor que reflete sua altura.

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

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