python - 如何使用 matplotlib 绘制类似 gnuplot 的 'with impulses' 的样式？

Question

我想用 python/pandas/matplotlib 创建一个如下图所示的图。上面的剪辑没问题，但我一直没能像下面的剪辑一样工作。我可以在 gnuplot 中做到这一点，其中等效的绘图风格是“带有冲动”。 matplotlib 可以吗？如果不能使用 matplotlib，是否有另一个 python 图形包可以工作？

Answer 1

创建这样一个图的最简单方法是使用 pyplot.stem . 可以找到一个例子here .

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0.1, 6*np.pi, 50)
plt.stem(x, np.cos(x)+1, linefmt='g-', markerfmt=' ')
plt.stem(x, -np.sin(x)-1, linefmt='r-', markerfmt=' ', basefmt="gray")

plt.show()

另一种选择是使用 pyplot.vlines .