python - カラーマップを使用してmplot3Dで4Dデータを表現する

Question

mplot3dサーフェスプロットでカラーマップが関連付けられている値を変更する方法はありますか？
例として、オブジェクトの表面温度を表現しようとしています。

import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

z = np.array([0,1,2,3,4,5,6,7,8,9,10])
radius = np.array([0,1,1.5,1,0,2,4,5,4,2,1])
temp = np.array([150,200,210,220,225,220,195,185,160,150,140])

angle = np.linspace(0,2*np.pi,20)
Z,ANG = np.meshgrid(z,angle)
# transform them to cartesian system
X,Y = radius*np.cos(ANG),radius*np.sin(ANG)

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='jet')
plt.show()

これにより、オブジェクトの3D表現が生成されますが、カラーマップはデフォルトでz軸の値に関連付けられています。カラーマップを「temp」値に関連付けることはできますか？

（この例では、「temp」は「radius」の値と同じ方法でZにマップされます）

私はMayaVIのようなツールを知っていますが、可能であれば、matplotlib内のソリューションを望んでいます。

Answer 1

plot_surfaceの呼び出しでfacecolorsを使用してみてください：

import matplotlib.pyplot as plt
import numpy as np

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

z = np.array([0,1,2,3,4,5,6,7,8,9,10])
radius = np.array([0,1,1.5,1,0,2,4,5,4,2,1])
temp = np.array([150,200,210,220,225,220,195,185,160,150,140])

angle = np.linspace(0,2*np.pi,20)
Z,ANG = np.meshgrid(z,angle)
T,ANG = np.meshgrid(temp,angle)
# transform them to cartesian system
X,Y = radius*np.cos(ANG),radius*np.sin(ANG)

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors=cm.jet(T/float(T.max())))
plt.show()