如何在Pandas分组时间序列中按0/1列值设置水平条形图颜色

1. 分组时同步保留异常标记 #

首先需要按日期分组，除了计算Clicks的总和，还要明确标记该日期是否存在Anomaly=1的记录（只要该日期有任意一条异常数据，就判定为异常日期）：

import pandas as pd
import matplotlib.pyplot as plt

# 假设你的原始DataFrame名为df
grouped_df = df.groupby('Date').agg(
    total_clicks=('Clicks', 'sum'),
    has_anomaly=('Anomaly', lambda x: x.any())  # 检查该日期是否存在异常
).reset_index()

2. 生成匹配的颜色列表 #

根据分组后的has_anomaly字段，生成对应每个日期条形的颜色列表：

# 异常日期用红色，正常日期用蓝色
bar_colors = ['red' if flag else 'blue' for flag in grouped_df['has_anomaly']]

3. 绘制带颜色标记的水平条形图 #

调用barh()时传入生成的颜色列表，确保每个条形的颜色和日期的异常状态对应：

plt.figure(figsize=(12, 8))
plt.barh(grouped_df['Date'].dt.strftime('%Y-%m-%d'), grouped_df['total_clicks'], color=bar_colors)

# 图表美化（可选）
plt.xlabel('总点击量')
plt.ylabel('日期')
plt.title('每日点击量统计（异常日期标记为红色）')
plt.tight_layout()
plt.show()

常见错误修正点 #

• 避免直接对Anomaly列求和后判断：求和结果大于1时逻辑判断可能出错，用any()能准确识别该组是否存在至少一条异常数据。
• 确保颜色列表长度和分组后的行数完全一致：如果分组后索引或行数有变动，会导致颜色和条形不匹配。

内容的提问来源于stack exchange，提问作者Mystic