python - 是否有 Matlab "stem3"的 matplotlib 副本？

Question

如 http://www.mathworks.com/help/techdoc/ref/stem3.html 中所述，使用 stem3 命令制作 3d 针状图非常容易。请问matplotlib中有没有类似的命令？我检查了最新版本的在线文档，但找不到。谁能给些建议？

Answer 1

我不知道在 matplotlib 中有任何与 stem3 直接等效的东西。然而，使用 Line3D 绘制这样的图形(至少是其基本形式)并不难。小号:

import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d.art3d as art3d
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, projection='3d')

N = 100
theta = np.linspace(0, 2*np.pi, N, endpoint=False)
x = np.cos(theta)
y = np.sin(theta)
z = range(N)
for xi, yi, zi in zip(x, y, z):        
    line=art3d.Line3D(*zip((xi, yi, 0), (xi, yi, zi)), marker='o', markevery=(1, 1))
    ax.add_line(line)
ax.set_xlim3d(-1, 1)
ax.set_ylim3d(-1, 1)
ax.set_zlim3d(0, N)    
plt.show()