plt.plot(inv_y,color='red',label='真实值') plt.plot(inv_y_predict,color='green',label='预测值') plt.xlabel('日期') plt.ylabel('收盘价') plt.title(title) plt.legend() plt.show()有14支股票,做一个循环
时间: 2023-08-11 12:06:28 浏览: 46
假设你有14支股票的真实值和预测值,可以将这些数据存储在一个字典中,字典的键为股票名称,值为真实值和预测值对应的列表,如下所示:
```
data = {
'stock1': {
'true_values': [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120, 130, 140],
'predicted_values': [12, 18, 33, 42, 48, 57, 68, 75, 88, 95, 108, 121, 138, 145]
},
'stock2': {
'true_values': [8, 16, 24, 32, 40, 48, 56, 64, 72, 80, 88, 96, 104, 112],
'predicted_values': [10, 19, 28, 36, 45, 54, 63, 72, 81, 90, 99, 108, 117, 126]
},
...
}
```
然后,你可以使用for循环遍历字典中的每个股票数据,并绘制对应的真实值和预测值对比图,如下所示:
```
import matplotlib.pyplot as plt
for stock_name, stock_data in data.items():
# 获取真实值和预测值列表
true_values = stock_data['true_values']
predicted_values = stock_data['predicted_values']
# 绘制对比图
title = f'{stock_name} Stock Price Comparison'
plt.plot(true_values, color='red', label='True Values')
plt.plot(predicted_values, color='green', label='Predicted Values')
plt.xlabel('Date')
plt.ylabel('Closing Price')
plt.title(title)
plt.legend()
plt.show()
```
运行以上代码,就可以绘制出14支股票的真实值和预测值对比图。你可以根据实际情况修改代码中的股票数据,以得到你需要的对比图。