将matplotlib视图设置为垂直于Python中的xy平面


问题内容

此处找到的代码是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()

和产量

在此处输入图片说明

有没有一种方法可以设置绘图视图,使其完全垂直于xy轴?这基本上将3-d图变成了2-d图,您可以在其中使用色图来判断z变量的大小,而不是其从z =
0基准的位移。


问题答案:

您想要的是ax.view_init带有的函数elev=90。看到这个答案

编辑:

添加ax.view_init(azim=0, elev=90)到您的脚本后,我得到以下信息:

在此处输入图片说明