用python画AR模型时序图

最新推荐文章于 2024-09-15 23:33:09 发布

潜水的飞鱼baby

最新推荐文章于 2024-09-15 23:33:09 发布

阅读量4.5k

点赞数 1

分类专栏： python_numpy 时间序列分析文章标签： python 图片

本文链接：https://blog.csdn.net/u011702002/article/details/78137497

版权

python_numpy 同时被 2 个专栏收录

25 篇文章

订阅专栏

时间序列分析

3 篇文章

订阅专栏

本文介绍使用Python绘制AR(自回归)模型时序图的方法。包括平稳与非平稳AR(1)及AR(2)模型，并展示了具体的实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

背景：

用python画AR模型的时序图。

结果：

代码：

import numpy as np
import matplotlib.pyplot as plt
"""
AR(1)的时序图：x[t]=a*x[t-1]+e
"""
num = 2000
e = np.random.rand(num)
x = np.empty(num)

"""
平稳AR(1)
"""
a = -0.5
x[0] = 2
for i in range(1,num):
    x[i] = a*x[i-1]+e[i]
plt.subplot(321,title = "AR({0}):x[t]={1}*x[t-1]+e".format(1,a))
plt.plot(x,"or")

"""
非平稳AR(1)
"""
a = -1.01
x[0] = 2
for i in range(1,num):
    x[i] = a*x[i-1]+e[i]
plt.subplot(322,title = "AR({0}):x[t]={1}*x[t-1]+e".format(1,a))
plt.plot(x,".b")

"""
平稳AR(2)
"""
a = -0.2
b = 0.7
x[0] = 2
for i in range(2,num):
    x[i] = a*x[i-1]+b*x[i-2]+e[i]
plt.subplot(323,title = "AR({0}):x[t]={1}*x[t-1]+{2}*x[t-2]+e".format(2,a,b))
plt.plot(x,"og")

"""
非平稳AR(2)
"""
a = -0.3
b = 0.8
x[0] = 2
for i in range(2,num):
    x[i] = a*x[i-1]+b*x[i-2]+e[i]
plt.subplot(324,title = "AR({0}):x[t]={1}*x[t-1]+{2}*x[t-2]+e".format(2,a,b))
plt.plot(x,".y")

"""
非平稳AR(2)
"""
a = -0.2
b = 0.8
x[0] = 2
for i in range(2,num):
    x[i] = a*x[i-1]+b*x[i-2]+e[i]
plt.subplot(313,title = "AR({0}):x[t]={1}*x[t-1]+{2}*x[t-2]+e".format(2,a,b))
plt.plot(x,"+",color="purple")

plt.show()