用matplotlib在3d中繪製imshow（）圖像

Question

如何在3d軸中繪製imshow()圖像？我試着用這個post。在那篇文章中，曲面圖與imshow()圖相同，但事實上並非如此。爲了說明，這裏我把不同的數據：

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

# create a 21 x 21 vertex mesh 
xx, yy = np.meshgrid(np.linspace(0,1,21), np.linspace(0,1,21)) 

# create vertices for a rotated mesh (3D rotation matrix) 
X = xx 
Y = yy 
Z = 10*np.ones(X.shape) 

# create some dummy data (20 x 20) for the image 
data = np.cos(xx) * np.cos(xx) + np.sin(yy) * np.sin(yy) 

# create the figure 
fig = plt.figure() 

# show the reference image 
ax1 = fig.add_subplot(121) 
ax1.imshow(data, cmap=plt.cm.BrBG, interpolation='nearest', origin='lower', extent=[0,1,0,1]) 

# show the 3D rotated projection 
ax2 = fig.add_subplot(122, projection='3d') 
ax2.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors=plt.cm.BrBG(data), shade=False) 

這裏是我的陰謀：

Answer 1

我認爲在3D VS 2D表面顏色你的錯誤是由於在表面的顏色數據標準化。如果您將傳遞至plot_surface facecolor的數據標準化爲facecolors=plt.cm.BrBG(data/data.max())，則結果更接近您的預期。

如果你只是想正常的座標，而不是使用imshow軸，片，你可以使用contourf，這是在3D支持作爲matplotlib 1.1.0，

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

# create a 21 x 21 vertex mesh 
xx, yy = np.meshgrid(np.linspace(0,1,21), np.linspace(0,1,21)) 

# create vertices for a rotated mesh (3D rotation matrix) 
X = xx 
Y = yy 
Z = 10*np.ones(X.shape) 

# create some dummy data (20 x 20) for the image 
data = np.cos(xx) * np.cos(xx) + np.sin(yy) * np.sin(yy) 

# create the figure 
fig = plt.figure() 

# show the reference image 
ax1 = fig.add_subplot(121) 
ax1.imshow(data, cmap=plt.cm.BrBG, interpolation='nearest', origin='lower', extent=[0,1,0,1]) 

# show the 3D rotated projection 
ax2 = fig.add_subplot(122, projection='3d') 
cset = ax2.contourf(X, Y, data, 100, zdir='z', offset=0.5, cmap=cm.BrBG) 

ax2.set_zlim((0.,1.)) 

plt.colorbar(cset) 
plt.show() 

雖然這不會在3D中的任意位置工作，其中imshow solution更好。