Python 折线图：可视化预测房价

使用 Python 折线图可视化预测房价

本教程将引导您使用 Python 的 matplotlib 库绘制折线图，以直观地展示真实房价和预测房价之间的差异。

代码示例

import matplotlib.pyplot as plt

def broken_line(y, y_pred, title):
    plt.figure()
    plt.plot(y, color='r', marker='o', label='真实房价')
    plt.plot(y_pred, color='b', marker='*', label='预测房价')
    plt.xlabel('房子数量')
    plt.ylabel('房子总价')
    plt.title(title)
    plt.legend()
    plt.show()

代码解释

导入 matplotlib 库：
```
import matplotlib.pyplot as plt
```
定义 broken_line 函数：
```
def broken_line(y, y_pred, title):
    # ...
```
- y: 真实房价数据列表
- y_pred: 预测房价数据列表
- title: 图表标题
创建图表：
```
plt.figure()
```
绘制真实房价折线：
```
plt.plot(y, color='r', marker='o', label='真实房价')
```
- color='r': 设置红色线条
- marker='o': 设置圆形标记
- label='真实房价': 设置图例标签
绘制预测房价折线：
```
plt.plot(y_pred, color='b', marker='*', label='预测房价')
```
- color='b': 设置蓝色线条
- marker='*': 设置星形标记
- label='预测房价': 设置图例标签

设置坐标轴标签：

plt.xlabel('房子数量')
plt.ylabel('房子总价')

设置图表标题：
```
plt.title(title)
```
添加图例：
```
plt.legend()
```
显示图表：
```
plt.show()
```

使用方法

将您的真实房价数据存储在 y 列表中。
将您的预测房价数据存储在 y_pred 列表中。
运行 broken_line(y, y_pred, '房价预测结果')，其中 '房价预测结果' 是您想要的图表标题。

示例

# 假设真实房价数据
y = [100, 120, 150, 180, 200]

# 假设预测房价数据
y_pred = [110, 130, 140, 190, 210]

broken_line(y, y_pred, '房价预测结果')

总结

通过绘制折线图，您可以直观地比较真实房价和预测房价，帮助您评估预测模型的性能。您可以根据需要调整代码，添加更多数据，修改图表样式，以满足您的分析需求。