如何更改colormap作用于哪个轴

Question

在matplotlib中绘制3D图形时，如何设置colormap以对y值进行操作而不是默认情况下的z值？

例如，在我绘图的那一刻，颜色贴图在z轴上起作用，但是我希望它使用y轴代替，我该怎么做？

箭头显示我想要颜色的方向。

Answer 1

作为一种解决方法，您可以切换轴并相应地设置视图：

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')

X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
Z = np.sin(np.sqrt(X*X + Y*Y))

surf = ax.plot_surface(X, Z, Y, rstride=1, cstride=1, cmap=cm.coolwarm, linewidth=0, antialiased=False)

ax.view_init(elev=120., azim=-90)
fig.colorbar(surf)

但这并不完美，但可能有更好的方法来设置视图。