前言
这是Python数据可视化的系列短篇,针对初级和中级用户,将理论和示例代码相结合,使用matplotlib, seaborn, plotly等工具创建不同主题的图表。
本文主题是如何在图表中显示误差。
import os
import warnings
import requests
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from statsmodels.tsa.arima_model import ARIMA
%matplotlib inline
plt.style.use("fivethirtyeight")
warnings.filterwarnings("ignore")
1. 显示测量误差
科学测量往往存在误差,准确的报告误差跟报告测量结果同样重要。
测量误差(也称为观察误差)是测量值和真实值之间的差异,包括随机误差(任何实验中都会自然发生的误差)和系统误差(由影响所有测量的错误校准的仪器引起)。
举一个例子,假设您正在测量100个学生的体重,有的学生穿较多的衣服,导致重500克,有的学生可能脱水导致轻200克,这些都是测量误差。事实上所有收集的样本都会存在误差,大多数情况下它们不可避免。所以在数据可视化中,有效显示这些错误可以传达更完整的信息。
在matplotlib中通过调用ax.errorbar()创建误差图(errorbar)。误差图由一个点和穿过这个点的线段组成,点对应测量结果,线段代表测量误差。
# 设置随机数种子
np.random.seed(123)
# 观测值的数量
n_observations = 30
x = np.arange(n_observations)
# 虚拟测量值
y = np.sin(x) + np.random.randn(n_observations)
# 测量误差,假设误差服从均匀分布
errors = np.random.randint(1, 3, n_observations)
# 调用ax.errorbar接口,提供3个数组,x, y, yerr
# fmt控制点和线段的样式,跟ax.plot的fmt相同
# color: 控制点的颜色
# ecolor: 代表误差的线段的颜色
# elinewidth: 代表误差的线段的大小
fig, ax = plt.subplots(figsize=(10, 7))
ax.errorbar(x, y, yerr=errors, fmt="o", color="black",
ecolor="lightgray", elinewidth=3)
2. 显示预测误差
在机器学习领域,回归模型和时间序列模型用于预测,模型会同时生成预测误差(所谓置信区间),在可视化预测结果时,通常要求显示预测值和预测区间,这可以通过ax.plot()和ax.fill_between()实现。
接下来使用一个真实的案例说明,先从蜂鸟数据获取黄金价格,然后用ARIMA(自回归整合移动平均)预测未来30天的价格,可视化预测结果。
# 从蜂鸟数据下载黄金价格
params = {
"symbol": "XAUUSD", # 现货黄金ID
"start_date": "2010-01-01",
"end_date": "2020-08-05",
"apikey": os.getenv("TROCHIL_API") # 在蜂鸟数据官网注册获取API密钥
}
resp = requests.get("https://api.trochil.cn/v1/forex/history", params)
data = pd.DataFrame.from_records(resp.json()["data"])
data["datetime"] = pd.to_datetime(data["datetime"])
data.set_index("datetime", inplace=True)
# 创建ARMIA模型,预测未来30天价格走势
model = ARIMA(data.close, order=(1, 1, 1))
results = model.fit()
fc, errors, conf = results.forecast(steps=30, alpha=0.05)
fc_df = pd.DataFrame({
"forecast": fc,
"lower": conf[:, 0],
"upper": conf[:, 1]
}, index=pd.date_range("2020-08-06", periods=30))
# 可视化预测结果,包括点预测值和预测区间
# 预测区间包括置信下限和上限,代表真实值可能落入的范围
# 调用ax.plot显示点预测值,ax.fill_between显示预测区间
data_since_2020 = data["2020":]
fig, ax = plt.subplots(figsize=(12, 7))
data_since_2020.close.plot(ax=ax, label="Historical Prices")
ax.plot(fc_df.index, fc_df.forecast, label="Forecast Prices")
ax.fill_between(fc_df.index, fc_df.lower, fc_df.upper, color="gray", alpha=0.3, label="Forecast Interval")
ax.set_title("Plot Forecast Errors")
ax.legend(loc="upper left")
<matplotlib.legend.Legend at 0x7fbfa348d9a0>
本文的文字及图片来源于网络,仅供学习、交流使用,不具有任何商业用途,版权归原作者所有,如有问题请及时联系我们以作处理。
作者:蜂鸟数据Trochil