如何使用matplotlib在数据帧中的两点之间绘制线

Question

我正在尝试使用matplotlib比较我的预测输出和测试数据。由于我是python的新手，我不知道如何像这张照片中那样用行连接每个条目。我可以写这样的代码来比较Y坐标和条目，但我无法用线将测试数据的每个条目与预测输出进行映射

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")

事实上，为了重现您想要的绘图，output和Y_test需要不同的x(例如，不同的X_1和X_2(。