matplotlib中的茎图？

Question

我想要plot(x, sin(x))，但不是从(xi,yi) 到(x_i+1,y_i+1) 我想要一条垂直线到从(xi,0) 到(xi,yi) 的每个点，因为有时点之间的插值没有意义（例如量化数据）（而且它看起来比没有垂直线要好得多）。

很像下面的茎图（来自 matlab 文档），但可以选择关闭圆圈和点之间的“-”。 不幸的是，我自己在 matplotlib 文档中找不到合适的绘图函数。

Answer 1

matplotlib 中也有一个stem plot（又名棒棒糖图）。

---

您可以在下面找到来自the docs 的示例。

import matplotlib.pyplot as plt
import numpy as np

# returns 10 evenly spaced samples from 0.1 to 2*PI
x = np.linspace(0.1, 2 * np.pi, 10)

markerline, stemlines, baseline = plt.stem(x, np.cos(x), '-.')

# setting property of baseline with color red and linewidth 2
plt.setp(baseline, color='r', linewidth=2)

plt.show()

---

如果要调整基线的y位置，可以使用bottom参数。

（改编）示例：

import matplotlib.pyplot as plt
import numpy as np

# returns 10 evenly spaced samples from 0.1 to 2*PI
x = np.linspace(0.1, 2 * np.pi, 10)

plt.stem(x, np.cos(x), '-.', bottom=-2)

plt.show()

Answer 2

我想你想使用 linestyle='steps--'

plot(x, sin(x), linestyle='steps--')