python贴图_python – 将颜色贴图应用于Matplotlib 3D曲面中的自定义轴

你可以试试这个：

import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

import numpy as np

X = np.linspace(-np.pi, np.pi, 200, endpoint=True)

Y = np.linspace(-np.pi, np.pi, 200, endpoint=True)

XX, YY = np.meshgrid(X,Y)

Z = np.cos(XX)*np.cos(YY)

fig = plt.figure()

ax1 = plt.subplot2grid((1,2), (0,0), projection='3d')

ax2 = plt.subplot2grid((1,2), (0,1), projection='3d')

surf = ax1.plot_surface(XX, YY, Z,

cmap=plt.cm.Set2)

surf2 = ax2.plot_surface(XX, YY, Z, shade=False,

facecolors=plt.cm.Set2((XX-XX.min())/(XX.max()-XX.min()))

)

在第二个图上,您将facecolors设置为XX的函数,而不是默认情况下的Z.您需要在0和1之间重新缩放XX值,否则色彩图将在0和1之外饱和.您还需要删除使用cmap时删除的阴影(在第一个图中).

然而,由于某些未知的原因,线条消失了.

您可以将它们添加回来：

plt.draw() # you need this to get the edge color

lines = np.array(surf2.get_edgecolor())

surf2.set_edgecolor(lines*np.array([0,0,0,0])+1) # make lines white, and keep alpha==1. It's an array of colors like this: [r,g,b,alpha]

它给：

HTH