python - 在 Python pandas autocorrelation_plot 中设置滞后数

Question

代码

import numpy as np
from pandas.tools.plotting import autocorrelation_plot
import matplotlib.pyplot as plt
nobs = 10000
xx = np.random.normal(size=nobs)
autocorrelation_plot(xx)
plt.show()

绘制 xx 的自相关，但它绘制了所有 10000 个滞后。如何只绘制前 10 个？

函数autocorrelation_plot 开始如下:

def autocorrelation_plot(series, ax=None, **kwds):
    """Autocorrelation plot for time series.
    Parameters:
    -----------
    series: Time series
    ax: Matplotlib axis object, optional
    kwds : keywords
        Options to pass to matplotlib plotting method

有没有办法使用 **kwds 参数设置绘制的滞后数？

Answer 1

autocorrelation_plot 返回一个 matplotlib.axis 对象。因此，您可以简单地使用 set_xlim() 方法来限制 x 轴:

autocorrelation_plot(xx).set_xlim([0, 10])

Reference