statsmodels.tsa.tsatools.add_lag¶

statsmodels.tsa.tsatools.add_lag(x, col=None, lags=1, drop=False, insert=True)[source]¶

返回一个包含滞后的数组，给定一个数组。

Parameters:¶

xarray_like

一个数组或NumPy ndarray子类。可以是具有列中观测值的1d或2d数组。

colint or None

col 可以是零基列索引的整数。如果它是一个一维数组，col 可以是 None。

lagsint

所需的滞后阶数。

dropbool

是否保留数据的同期变量。

insertbool or int

如果为True，则在col之后插入滞后值。如果为False，则附加数据。如果为int，则在int处插入滞后值。

Returns:¶

arrayndarray

带有滞后项的数组

注释

修剪数组的前后部分，使得返回的数组长度为 len(X) - lags。滞后值按递增顺序返回，即 t-1, t-2, …, t-lags

示例

>>> import statsmodels.api as sm
>>> data = sm.datasets.macrodata.load()
>>> data = data.data[['year','quarter','realgdp','cpi']]
>>> data = sm.tsa.add_lag(data, 'realgdp', lags=2)