web-dev-qa-db-ja.com

matplotlibの3Dでのimshow（）画像のプロット

imshow()画像を3D軸にプロットする方法は？私はこれを試していました 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)

これが私のプロットです：

http://www.physics.iitm.ac.in/~raj/imshow_plot_surface.png

pythonmatplotlib3d
9
Raj

3Dと2Dの表面色の誤差は、表面色のデータの正規化が原因だと思います。 _plot_surface_ facecolorに渡されたデータをfacecolors=plt.cm.BrBG(data/data.max())で正規化すると、結果は期待したものに近づきます。

imshowを使用する代わりに、座標軸に垂直なスライスが必要な場合は、contourfを使用できます。これは、matplotlib 1.1.0以降で3Dでサポートされています。

_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()
_

このコードはこの画像になります：

result

これは、imshow solution の方が良い3Dの任意の位置にあるスライスでは機能しません。

8
Ed Smith