matplotlib：在2d plot

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的doc中找不到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)

虽然这似乎并不是您所需要的，但这是朝着正确方向迈出的一步。