如何使用Matplotlib在时间序列图中高亮指定索引范围的子序列
使用Matplotlib高亮时间序列中的指定子序列
要实现时间序列中指定子序列的颜色高亮,你可以将序列拆分为普通部分和高亮部分分别绘制,或者在整体序列之上叠加高亮子序列。以下是具体实现步骤:
1. 准备基础代码与数据
先导入所需库,并准备你的时间序列数据(示例用随机生成数据替代,可替换为真实数据):
import matplotlib.pyplot as plt import numpy as np # 生成示例时间序列(替换为你的真实数据) np.random.seed(42) data = np.cumsum(np.random.randn(300)) x = np.arange(len(data)) # 定义高亮区域参数 start_idx = 203 window_length = 40 end_idx = start_idx + window_length
2. 基础高亮实现(仅子序列变色)
先绘制整个序列作为底色,再单独绘制高亮子序列,用更显眼的颜色和线宽突出:
plt.figure(figsize=(12, 6)) # 绘制整体时间序列(灰色底色) plt.plot(x, data, color="#999999", label="整体序列") # 绘制高亮子序列(红色粗线) plt.plot(x[start_idx:end_idx], data[start_idx:end_idx], color="#ff3333", linewidth=2, label="高亮子序列") # 添加图表标签与图例 plt.xlabel("索引") plt.ylabel("序列值") plt.title("时间序列子序列高亮") plt.legend() plt.show()
3. 增强版:背景+子序列双重高亮
如果需要更突出的视觉效果,可以同时用半透明色块高亮背景区域:
fig, ax = plt.subplots(figsize=(12, 6)) # 绘制整体序列 ax.plot(x, data, color="#999999") # 添加背景高亮色块(半透明粉色) ax.axvspan(start_idx, end_idx, color="#ffcccc", alpha=0.3) # 叠加高亮子序列 ax.plot(x[start_idx:end_idx], data[start_idx:end_idx], color="#ff3333", linewidth=2) # 设置图表属性 ax.set_xlabel("索引") ax.set_ylabel("序列值") ax.set_title("时间序列高亮(背景+子序列)") plt.show()
核心逻辑是通过截取指定索引范围的子序列,用不同样式重新绘制在整体序列之上,实现局部高亮效果。
内容的提问来源于stack exchange,提问作者Aditya Bhandwalkar
相关产品推荐
相关产品推荐

