本文共 1420 字,大约阅读时间需要 4 分钟。
安装必要库确保你已安装 plotly 和 pandas 库。如果尚未安装,可以运行以下命令安装:
pip install plotly pandas
导入开发库在 Python 环境中导入 Plotly Express 和 pandas:
import plotly.express as pximport 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 组合散点图和线图的完整指南,适用于多种数据分析场景。
转载地址:http://lgtfk.baihongyu.com/