gpt4 book ai didi

python - 修改 statsmodels 图

转载 作者:行者123 更新时间:2023-12-04 08:31:13 26 4
gpt4 key购买 nike

我在这里关注 statsmodels 文档:
https://www.statsmodels.org/stable/vector_ar.html
我到了页面中间的部分,上面写着:irf.plot(orth=False)它为我的数据生成以下图表:
enter image description here
我需要修改图表的元素。例如,我需要应用 tiny_layout 并减小 y-tick 大小,这样它们就不会进入左边的图表。
文档讨论了将“子图绘图函数”传递给 subplot irf.plot() 的参数。但是当我尝试类似的东西时:

irf.plot(subplot_params = {'fontsize': 8, 'figsize' : (100, 100), 'tight_layout': True})
只有 fontsize参数工作。我也尝试将这些参数传递给 'plot_params' 参数,但无济于事。
所以,我的问题是如何访问这个 irf.plot 的其他参数,尤其是 figsize 和 ytick 大小?我还需要强制它打印一个网格,以及 x 轴上的所有值 (1, 2, 3, 4, ..., 10)
有什么方法可以使用 fig, ax = plt.subplots() 创建一个空白图然后在那个图上创建 irf.plot ?

最佳答案

看起来该函数返回一个 matplotlib.figure :
尝试这样做:

fig = irf.plot(orth=False,..)
fig.tight_layout()
fig.set_figheight(100)
fig.set_figwidth(100)
如果我用这个例子运行它,它会起作用:
import numpy as np
import pandas
import statsmodels.api as sm
from statsmodels.tsa.api import VAR

mdata = sm.datasets.macrodata.load_pandas().data
dates = mdata[['year', 'quarter']].astype(int).astype(str)
quarterly = dates["year"] + "Q" + dates["quarter"]
from statsmodels.tsa.base.datetools import dates_from_str
quarterly = dates_from_str(quarterly)
mdata = mdata[['realgdp','realcons','realinv']]
mdata.index = pandas.DatetimeIndex(quarterly)
data = np.log(mdata).diff().dropna()
model = VAR(data)

results = model.fit(maxlags=15, ic='aic')
irf = results.irf(10)

fig = irf.plot(orth=False)
fig.tight_layout()
fig.set_figheight(30)
fig.set_figwidth(30)
enter image description here

关于python - 修改 statsmodels 图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/65014593/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com