def report(self): # plt.plot(self.solution[:,0],color = 'darkblue',label = 'Susceptible',marker = '.') plt.plot(self.solution[:, 1], color='orange', label='Infection', marker='.') plt.plot(self.solution[:, 2], color='green', label='Recovery', marker='.') plt.title('SIR Model' + ' infectionProb = ' + str(infectionProb)) plt.legend() plt.xlabel('Day') plt.ylabel('Number of people') plt.show() 这段代码有没有问题
时间: 2024-02-14 08:13:41 浏览: 60
这段代码有问题,因为它只画了感染者和康复者的数量变化曲线,而没有画易感者的数量变化曲线。根据SIR模型,易感者的数量应该随时间逐渐减少,感染者的数量应该在一定时期内快速增加,然后逐渐减少,康复者的数量应该逐渐增加。因此,应该添加一个易感者的数量变化曲线,以便全面呈现SIR模型的结果。可以在代码中添加以下内容:
```
plt.plot(self.solution[:,0], color='blue', label='Susceptible', marker='.')
```
这样就可以画出易感者的数量变化曲线了。
阅读全文