在Python中设置matplotlib视图垂直于x-y平面

bfhwhh0e 于 2023-10-24 发布在 Python

关注(0)|答案(2)|浏览(125)

下面的代码是一个3D曲面图的例子：

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
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)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
        linewidth=0, antialiased=False)
ax.set_zlim(-1.01, 1.01)

ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

和产量

有没有一种方法可以设置绘图视图，使其完全垂直于x-y轴？这基本上可以将三维绘图转换为二维绘图，您可以使用色彩Map来判断z变量的大小，而不是其相对于z=0基准的位移。

matplotlib

来源：https://stackoverflow.com/questions/33084853/set-matplotlib-view-to-be-normal-to-the-x-y-plane-in-python

2条答案

按热度按时间

sbtkgmzw1#

您需要的是ax.view_init函数，其中包含elev=90。参见this answer
编辑：
在将ax.view_init(azim=0, elev=90)添加到您的脚本之后，我得到了以下内容：

赞(0）回复(0）举报 2023-10-24

des4xlb02#

你需要pcolor来实现：

import matplotlib.pyplot as plt
import numpy as np

dx, dy = 0.25, 0.25

y, x = np.mgrid[slice(-5, 5 + dy, dy),
                slice(-5, 5 + dx, dx)]
R = np.sqrt(x**2 + y**2)
z = np.sin(R)

z = z[:-1, :-1]
z_min, z_max = -np.abs(z).max(), np.abs(z).max()


plt.subplot()
plt.pcolor(x, y, z, cmap='RdBu', vmin=z_min, vmax=z_max)

plt.axis([x.min(), x.max(), y.min(), y.max()])
plt.colorbar()
plt.show()

其他演示是here

赞(0）回复(0）举报 2023-10-24

我来回答

在Python中设置matplotlib视图垂直于x-y平面

2条答案

相关问题

热门标签

最新问答