如何在python中绘制实际值与预测值的关系图？

Question

          Original      Predicted
0           6            1.56
1           12.2         3.07
2           0.8          2.78
3           5.2          3.54
.                    

我尝试过的代码：

def plotGraph(y_test,y_pred,regressorName):
    if max(y_test) >= max(y_pred):
        my_range = int(max(y_test))
    else:
        my_range = int(max(y_pred))
    plt.scatter(y_test, y_pred, color='red')
    plt.plot(range(my_range), range(my_range), 'o')
    plt.title(regressorName)
    plt.show()
    return

我想要的图表：

但我目前的产出是：

Answer 1

在matplotlib（我假设您正在使用它）中documentation有一个matplotlib.pyplot.scatter函数的信息，前两个参数是：

x, y : float or array-like, shape (n, )
The data positions.

因此，对于您的应用程序，您需要在同一个图形上绘制两个散点图-使用matplotlib.pyplot.scatter两次。首先是y_test作为y和color='red'，其次是y_pred作为y和color='blue'

不幸的是，您没有为y_test和y_pred提供x值的信息，但是您也需要这些信息来定义x函数调用中的plt.scatter

绘制两个散点图有点棘手，正如this answer所说，它需要对Axes对象的引用。例如（如答案所示）：

import matplotlib.pyplot as plt

x = range(100)
y = range(100,200)
fig = plt.figure()
ax1 = fig.add_subplot(111)

ax1.scatter(x[:4], y[:4], s=10, c='b', marker="s", label='first')
ax1.scatter(x[40:],y[40:], s=10, c='r', marker="o", label='second')

有关更多详细信息，请查看matplotlib文档和提到的答案