matplotlib：在2d图上投影3d曲面

Question

在2D中是否有等效于“ Axes3DSubplot.plot_surface”的文件？ 我正在尝试在matplotlib中绘制网格在XY平面上的投影（因此不在“ 3d”模式下）。

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

# Data (wireframe sphere)
theta, phi = np.meshgrid( np.linspace(0, np.pi/2, 10), np.linspace(0, np.pi/2, 10) )

x = np.sin(theta) * np.cos(phi)
y = np.sin(theta) * np.sin(phi)
z = np.cos(theta)


fig = plt.figure()

# Subplot 3D
ax1 = fig.add_subplot(1, 2, 1, projection='3d', aspect='equal')

colors = matplotlib.cm.jet(np.hypot(x,y))

surface = ax1.plot_surface(x, y, z, rstride=1, cstride=1, facecolors = colors, alpha=0.5 )
projection = ax1.plot_surface(0, y, z, rstride=1, cstride=1, facecolors = colors )
projection.set_edgecolor('k')

# Subplot 2D
ax2 = fig.add_subplot(1, 2, 2, aspect='equal')

ax2.plot(y, z, 'k')
ax2.plot(y.T, z.T, 'k')

我正在尝试产生类似于以下结果：

ax1.plot_surface(0, y, z, rstride=1, cstride=1, facecolors = colors )

但是在2D子图中。 我在AxesSubplot的文档中找不到plot_surface的等效项。 我唯一要做的是使用以下方法绘制线框（但不是面部颜色）：

ax2.plot(y, z, 'k')
ax2.plot(y.T, z.T, 'k')

我无法上传图像，但基本上，我想将“颜色”放在第二个子图中。 谢谢，

编辑：@Tim是的，我想在这种情况下，我设法做到了：

ax2.contourf(y, z, np.hypot(x,y), levels=np.hypot(x,y)[0], cmap=matplotlib.cm.jet)

在更通用的情况下，您将需要正确的色阶功能以及对色阶和色彩图的一些调整，但这似乎是可行的。

另一个解决方案是使用matplotlib.patches.Polygon绘制每个投影面。

Answer 1

您可以使用contourf生成彩色的2D轮廓：

# Subplot 2D
ax2 = fig.add_subplot(1, 2, 2, aspect='equal')
ax2.contourf(x, y, z, facecolors=colors)

尽管这似乎并不是您真正需要的，但这是朝着正确方向迈出的一步。