将matplotlib中干图的基线延伸到轴极限

使用matplotlib生成干图时，如何使基线延伸到轴限制，而不是以最后的数据点结束

---

函数调用返回

plt.stem

绘图的基线

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

您可以将该行的数据设置为

[0,1]

，并使用

yaxis\u变换

，以便将这些坐标解释为轴坐标

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0.1, 2*np.pi, 10)
markerline, stemlines, baseline = plt.stem(x, np.cos(x), '-.')
plt.setp(baseline, 'color', 'r', 'linewidth', 2)

baseline.set_xdata([0,1])
baseline.set_transform(plt.gca().get_yaxis_transform())

plt.show()

---

请向您表示感谢-这很有效。对不起，我的第一篇文章写得不好，谢谢你整理。我来自MATLAB多年，最近刚刚开始尝试使用Python。你是怎么想出这个解决办法的？我找不到任何关于基线句柄属性的文档。您是对的，这个解决方案到目前为止并不明显，您只有在其他情况下使用了

get\u yaxis\u transform（）

才能知道它。简言之：经验。然而，stem plot返回三个对象的事实在文档中是清楚的。另外，查看返回类型并阅读文档中返回的对象通常也会有所帮助。