本文共 1445 字，大约阅读时间需要 4 分钟。

Plotly Express 组合散点图和线图指南

安装必要库 确保你已安装 plotly 和 pandas 库。如果尚未安装，可以运行以下命令安装：

pip install plotly pandas

导入开发库 在 Python 环境中导入 Plotly Express 和 pandas：

import plotly.express as px
import pandas as pd

创建数据集 创建包含 x 和 y 列的数据集：

data = {'x': [1, 2, 3, 4, 5], 'y': [2, 4, 6, 8, 10]}
df = pd.DataFrame(data)

绘制散点图并添加线图 使用 px.scatter() 绘制散点图，然后通过 add_traces() 方法添加线图：

fig = px.scatter(df, x='x', y='y')
line_data = {'x': [1, 5], 'y': [2, 10]}
line_df = pd.DataFrame(line_data)
fig.add_traces(px.scatter(line_df, x='x', y='y').data[0])

显示图表 将生成的图表显示出来：

fig.show()

调整样式与颜色 可以在 px.scatter() 中设置参数调整颜色或样式：

fig = px.scatter(df, x='x', y='y', color_discrete_sequence=['blue'])

保存为 HTML 文件 如果需要将图表保存为 HTML 文件：

fig.write_html('combined_plot.html')

测试用例 验证代码逻辑：

def test_combine_scatter_line():
    data = {'x': [1, 2, 3, 4, 5], 'y': [2, 4, 6, 8, 10]}
    df = pd.DataFrame(data)
    fig = px.scatter(df, x='x', y='y')
    line_data = {'x': [1, 5], 'y': [2, 10]}
    line_df = pd.DataFrame(line_data)
    fig.add_traces(px.scatter(line_df, x='x', y='y').data[0])
    assert len(fig.data) == 2
    print("测试通过！")

测试代码：

test_combine_scatter_line()

应用场景示例 如果数据集中包含分类列，可以按分类绘制散点图：

fig = px.scatter(df, x='x', y='y', color='category')
line_data = {'x': [1, 5], 'y': [2, 10]}
line_df = pd.DataFrame(line_data)
fig.add_traces(px.scatter(line_df).data[0])

为线图设置颜色：

fig.add_traces(px.scatter(line_df, color='blue').data[0])

以上步骤提供了使用 Plotly Express 组合散点图和线图的完整指南，适用于多种数据分析场景。