怎么解决上面的代码问题呢

关于变量 ax 没有被定义的问题，你可以将其定义为一个子图对象，例如：

fig, ax = plt.subplots(figsize=(10, 10))

关于 example_arma 模块的问题，你可以将其替换成其他的时间序列模块，例如：

from statsmodels.tsa.arima_process import arma_generate_sample

最后，如果你需要在 Jupyter notebook 中运行这段代码，可以在开头添加 `%matplotlib inline`。