如何在Python的plotnine中基于日期X轴创建geom_segment()图表

我明白你想给每个分类（category）添加一条横跨日期X轴的灰色背景线段，就像参考图里的效果。之前的代码出问题主要是因为没有正确设置线段的起止坐标映射，咱们一步步来解决：

问题根源 #

你之前的代码要么是用了原数据里的单个日期作为线段端点，要么是没有为每个分类单独指定横跨整个日期范围的线段数据，导致plotnine无法正确生成你想要的水平线段。

解决方案：创建辅助数据集 #

要实现每个分类对应一条横跨日期范围的线段，我们需要先构建一个辅助数据集，每个分类占一行，明确给出线段的起始和结束日期（通常是整个数据集的最早和最晚日期），同时固定y轴为分类本身（保证线段是水平的）。

下面是完整的可运行代码：

import pandas as pd
from plotnine import ggplot, aes, geom_segment, theme, element_text, geom_point

# 你的原始数据集
data = pd.DataFrame({
    'date': pd.date_range('2022-01-01', periods=12, freq="ME"),
    'category': ['A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'C'],
    'value': [10, 20, 30, 15, 25, 35, 20, 30, 40, 25, 35, 45]
})

# 构建辅助数据集：每个分类对应一条横跨整个日期范围的线段
seg_data = pd.DataFrame({
    'category': data['category'].unique(),  # 获取所有不重复的分类
    'x_start': data['date'].min(),          # 数据集的最早日期
    'x_end': data['date'].max()             # 数据集的最晚日期
})

# 绘制图表：先画背景线段，再叠加原始数据（这里用散点示例）
(ggplot(data)
 # 先画灰色背景线段，用辅助数据集
 + geom_segment(data=seg_data, aes(x='x_start', xend='x_end', y='category', yend='category'),
               size=8.5, color="#edece3")
 # 可选：叠加原始数据的可视化元素，比如散点
 + geom_point(aes(x='date', y='category', color='value'), size=3)
 # 调整X轴文本角度，避免重叠
 + theme(axis_text_x=element_text(angle=45, hjust=1))
)

进阶调整：按分类自身日期范围绘制线段 #

如果不想让线段横跨整个数据集的日期，而是只覆盖每个分类有数据的日期范围，可以用分组统计来生成辅助数据：

# 按分类分组，计算每个分类的最早和最晚日期
seg_data = data.groupby('category')['date'].agg(['min', 'max']).reset_index()
seg_data.columns = ['category', 'x_start', 'x_end']

把这段代码替换上面的辅助数据集生成部分，就能得到每个分类专属日期范围的背景线段了。

备注：内容来源于stack exchange，提问作者ViSa