如何在 Python 中绘制带有置信区间的时间序列数组?(Matplotlib)

matplotlibpythondata visualization

要在 Python 中绘制带有置信区间的时间序列数组,我们可以采取以下步骤 −

  • 设置图形大小并调整子图之间和周围的填充。
  • 获取时间序列数组。
  • 初始化变量 n_steps 以获取平均值和标准差。
  • 获取置信区间的下线和上线。
  • 使用 plot() 方法绘制均值线。
  • 使用 fill_between() 方法获取置信区间。
  • 要显示图形,请使用 show() 方法。

示例

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

time_series_array = np.sin(np.linspace
                           (-np.pi, np.pi, 400)) + np.random.rand((400))
n_steps = 15

time_series_df = pd.DataFrame(time_series_array)

line = time_series_df.rolling(n_steps).mean()

line_deviation = 2 * time_series_df.rolling(n_steps).std()

under_line = (line - line_deviation)[0]

over_line = (line + line_deviation)[0]

plt.plot(line, linewidth=2)

plt.fill_between(line_deviation.index, under_line,
                  over_line, color='red', alpha=.3)

plt.show()

输出


相关文章