Python Plotly分面图表添加动态形状 解决线条仅首分面显示问题

问题原因 #

Plotly添加形状时默认绑定第一个分面的坐标轴，没有指定对应分面的轴引用时，所有形状都会渲染在首个分面中。

解决思路 #

1. 先生成分面取值与对应坐标轴引用的映射关系：用facet_col生成的分面，x轴按顺序命名为x（首个分面）、x2（第二个分面）、x3以此类推，共享y轴时所有分面的y轴引用均为y。
2. 循环添加线条时，为每个线条指定对应分面的xref参数，绑定到正确的坐标轴。

修改后完整代码 #

import plotly.express as px

fig = px.scatter(
        x=df['x_variable'],
        y=df['y_variable'],
        facet_col = df['facet_variable'])

# 构造分面值到x轴引用的映射
facet_vals = df['facet_variable'].unique().tolist()
facet_xref_map = {val: f'x{i+1}' if i>0 else 'x' for i, val in enumerate(facet_vals)}

# 循环添加竖线
for i in range(len(df)):
    current_facet = df['facet_variable'].iloc[i]
    fig.add_shape(
        type="line",
        x0=df['x_variable'].iloc[i],
        y0=0,
        x1=df['x_variable'].iloc[i],
        y1=df['y_variable'].iloc[i],
        xref=facet_xref_map[current_facet],
        yref='y',
        line=dict(color='grey', width=2) # 可自定义线条颜色、宽度
    )

fig.show()

注意事项 #

• 如果使用facet_row生成分面，需要映射的是yref参数，命名规则为y、y2、y3以此类推
• 如果分面同时设置了独立y轴，需要按同样逻辑构造yref的映射关系
• 如需批量调整线条样式，可直接在add_shape的line参数中统一配置

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