Python matplotlib中从点到轴的虚线

我有一组点，我正在这样绘制：

import matplotlib.pyplot as plt`

x = [1,2,3,4,5,6]
y = [1,4,9,16,25,36]

plt.scatter(x, y)

plt.show()

这会产生这样的输出

我想要的是从点到轴的垂直度，如下图所示：

---

如何实现这一点？

您正在寻找的图被称为stemplot，它有一个您可以找到的答案。它将为您提供与所需x轴的垂直线连接；但是如果你想要水平线连接到y轴，可能需要一些特殊的处理。您可以在另一个旋转轴或其他轴上使用干图。

使用

hlines

和

vlines

可以分别绘制水平线和垂直线

import matplotlib.pyplot as plt

x = [1,2,3,4,5,6]
y = [1,4,9,16,25,36]

plt.vlines(x, 0, y, linestyle="dashed")
plt.hlines(y, 0, x, linestyle="dashed")
plt.scatter(x, y, zorder=2)

plt.xlim(0,None)
plt.ylim(0,None)
plt.show()

---

我正在寻找一个更通用的解决方案：例如，两个数组中的元素是否可以传递给一个函数，该函数将在任意点之间绘制虚线：在本例中，（x，y）到（x，0）表示垂直线，（x，y）到（0，y）表示水平线，以便涵盖这两种情况