如何在Plotly Express分组条形图中动态获取所有X坐标？

要解决分组柱状图中动态获取每个条形X坐标的问题，无需硬编码数值，可通过提取图表内置参数并计算偏移量实现，适配任意color参数下的分组数量变化。

解决方案代码 #

import plotly.express as px
import plotly.graph_objects as go
import numpy as np

df = px.data.tips()
# 可替换color参数（如"time"或"day"）测试动态适配
fig = px.histogram(
    df, x="sex", y="total_bill", color="day", barmode="group", height=400
)

# 提取图表关键参数
num_categories = len(fig.layout.xaxis.categoryarray)  # 主X轴类别数量
num_groups = len(fig.data)  # 每组内的条形数量（由color参数决定）
# 获取默认间隙参数（若未手动设置）
bargap = fig.layout.bargap if fig.layout.bargap is not None else 0.2
bargroupgap = fig.layout.bargroupgap if fig.layout.bargroupgap is not None else 0.1

# 计算每组内条形的位置偏移量
group_width = (1 - bargap) / num_groups  # 每个条形（含组内间隙）的宽度
offsets = np.linspace(-(num_groups-1)*group_width/2, (num_groups-1)*group_width/2, num_groups)

# 添加覆盖式辅助X轴
fig.layout.xaxis2 = go.layout.XAxis(
    overlaying='x', range=[-0.5, num_categories - 0.5], showticklabels=False
)

# 为每个条形添加辅助标记（示例用灰色虚线）
for cat_idx in range(num_categories):
    for group_idx in range(num_groups):
        bar_x = cat_idx + offsets[group_idx]
        # 将标记放在图表顶部，避免遮挡数据
        marker_y = [df['total_bill'].max() + 10]
        fig.add_scatter(
            x=[bar_x], y=marker_y, xaxis='x2',
            showlegend=False, line={'color': 'gray', 'dash': 'dash'}
        )

fig.show()

原理说明 #

1. 参数提取：从图表布局中获取主类别数量、分组数量，以及默认的bargap（类别间间隙）和bargroupgap（组内条形间隙）参数；
2. 偏移计算：基于间隙参数计算每组内每个条形相对于主类别中心位置的偏移量，确保无论分组数量如何变化，都能精准定位每个条形的X坐标；
3. 动态适配：遍历所有主类别和分组，自动生成每个条形的位置标记，无需修改代码即可适配不同的color参数配置。

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