如何使用 matplotlib 在 dataframe 中的两点之间画线？

Question

我正在尝试比较我预测的 output 和使用 matplotlib 的测试数据。由于我是 python 的新手，所以我无法找到如何将每个条目与这张照片中的行连接起来。 我能够编写这样的代码来比较 Y 坐标和条目，但我无法 map 测试数据的每个条目与预测的 output 与一行

X_1 = range(len(Y_test))
plt.figure(figsize=(5,5))
plt.scatter(X_1, output, label='Y_output',alpha=0.3)
plt.scatter(X_1, Y_test, label='Y_test',alpha=0.3)
plt.title("Scatter Plot")
plt.legend()
plt.xlabel("entries")
plt.ylabel("Y value")
plt.show()

我们得到的图表

Answer 1

除了你的代码之外，试试这样的东西

plt.plot(np.stack((X_1,X_1)), np.stack((output,Y_test)), color="black")

实际上，要重现您想要的 plot，您需要为output和Y_test提供不同的 x（例如，不同的X_1和X_2 ）。