首页 > 解决方案 > 如何在 Pandas 中绘制和标记多个自相关函数?

问题描述

我有几个变量,我想在一张图上查看它们的自相关函数。我可以做这个。但问题是我不确定如何创建一个图例,以便我知道哪个变量是哪个。

这是我的数据的样子:

import pandas as pd 
from pandas.plotting import autocorrelation_plot
df = pd.DataFrame(data = {
    "Year": [y for y in range(1800, 2000)],
    "Var 1": [random.random() for i in range(200)], 
    "Var 2": [random.random() for i in range(200)], 
    "Var 3": [random.random() for i in range(200)]
})

df.set_index("Year")

现在这是我在一张图上绘制自相关函数的方法:

for variable in df.columns:
    autocorrelation_plot(df[variable])

问题是没有传说,所以我不知道哪个变量是哪个。

而且,autocorrelation_plot没有legend论据。

标签: pythonpandastime-seriesautocorrelation

解决方案


I have added a label variable to get the labels:

for variable in df.columns:
   ax = autocorrelation_plot(df[variable], label = variable)

Stocks data

and it worked


推荐阅读