컬러 맵이 작동하는 축을 변경하는 방법

Question

matplotlib에 3D 그림을 그릴 때 기본적으로 수행되는 z 값 대신 y 값 대신 작동하도록 색상 맵을 설정하려면 어떻게해야합니까?

예를 들어 플로팅 할 때 컬러 맵이 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) 

그것은 비록 완벽하지 않지만보기를 설정하는 더 나은 방법이있을 수 있습니다.