要為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()
函數顯示出散點圖和趨勢線。您也可以根據需要自定義趨勢線的類型和樣式。