温馨提示×

Plotly怎么为散点图添加趋势线和回归线

小亿
91
2024-05-15 17:52:18
栏目: 编程语言

要为Plotly散点图添加趋势线和回归线,可以使用plotly.express库中的px.scatter()函数,然后使用px.get_trendline_results()函数来获取趋势线和回归线的数据。

下面是一个示例代码:

import plotly.express as px

# 创建散点图
df = px.data.iris()
fig = px.scatter(df, x="sepal_width", y="sepal_length", trendline="ols")

# 获取趋势线和回归线的数据
results = px.get_trendline_results(fig)
print(results)

# 显示图表
fig.show()

在上面的代码中,我们使用了Plotly自带的鸢尾花数据集来创建一个散点图,并指定了trendline="ols"参数来添加最小二乘回归线。然后使用px.get_trendline_results()函数来获取趋势线和回归线的数据,并将其打印出来。

最后,使用fig.show()函数显示出散点图和趋势线。您也可以根据需要自定义趋势线的类型和样式。

0