如何正确使用生成器在Plotly中绘制多条垂直线？

你的代码问题所在 #

你当前的列表推导式生成的x是包含多个元组的二维列表（比如[(date1,date1), (date2,date2)]），但Plotly的go.Scatter要求x和y必须是一维序列，且两者长度必须完全匹配。你的y只有2个元素，而x的长度等于high_peaks_dates的长度，长度不匹配导致代码无法正常运行。

能否使用生成器？ #

可以使用生成器，但需要生成符合要求的一维序列。生成器的作用是逐个产出元素，只要最终产出的x和y是长度匹配的一维序列即可。

最优替代方案：单Trace绘制多条垂直线 #

要在一个trace里完成所有垂直线的绘制（避免加载慢、图例重复的问题），需要构造正确的x和y序列：

• x序列：将每个日期重复两次，展开成一维列表（比如[date1, date1, date2, date2, ...]）
• y序列：将[最高值, 最低值]重复len(high_peaks_dates)次（比如[max, min, max, min, ...]）

这样Plotly会按顺序连接每一组(date, max)和(date, min)，形成一条垂直线，所有线条都属于同一个trace，图例只显示一次，加载效率也更高。

代码实现 #

import plotly.graph_objects as go

# 构造一维x序列：每个日期重复两次
x = [date for date in high_peaks_dates for _ in range(2)]
# 构造一维y序列：[max, min]重复对应次数
y = [df.High.max(), df.High.min()] * len(high_peaks_dates)

figure.add_trace(
    go.Scatter(
        x=x,
        y=y,
        mode='lines',
        name='Peak Vertical Lines',
        # 可选：自定义线条样式
        line=dict(color='crimson', dash='dash')
    )
)

如果想用生成器构造x，可以写成：

x = (date for date in high_peaks_dates for _ in range(2))

Plotly会自动将生成器转换为列表，效果和列表推导式一致。

特殊场景：每条垂直线样式不同 #

如果需要为不同的垂直线设置不同样式（比如颜色、粗细），且无法使用add_vline，可以通过循环添加trace，但控制图例只显示一次：

is_first_trace = True
for date in high_peaks_dates:
    # 自定义样式，示例：根据日期设置不同颜色
    line_color = 'red' if date.year < 2008 else 'blue'
    
    figure.add_trace(
        go.Scatter(
            x=[date, date],
            y=[df.High.max(), df.High.min()],
            mode='lines',
            name='Peak Lines',
            # 只让第一个trace显示图例
            showlegend=is_first_trace,
            line=dict(color=line_color, dash='dash')
        )
    )
    is_first_trace = False

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