yiskw note

機械学習やプログラミングについて気まぐれで書きます

plotly expressで散布図に色を変えた近似直線を追加する

python plotly

概要

最近plotly expressを用いてデータの可視化を行なっているのですが， 散布図に近似直線を加える方法とその直線の色を変える方法について調査したのでメモしておきます．

plotly expressのインストール方法

pipでインストールする場合，以下のコマンドを実行するだけです．

pip install plotly statsmodels

statsmodelsは，近似直線を算出する際に必要なライブラリみたいです．

plotly expressでの散布図を作成

plotly expressを用いると簡単に散布図を作成できます．

import plotly.express as px

# サンプルデータの読み込み
df = px.data.tips()
fig = px.scatter(df, x="total_bill", y="tip")
fig.show()

f:id:yiskw713:20210220132720p:plain

散布図に近似直線を追加

散布図に近似直線を追加するには，px.scatterの引数に，trendline="ols"を設定します． またその色を変更するには，trendline_color_overrideという引数に色を指定すれば良いです．

import plotly.express as px

# サンプルデータの読み込み
df = px.data.tips()
fig = px.scatter(
    df,
    x="total_bill",
    y="tip", 
    trendline="ols",
    trendline_color_override="red",
)
fig.show()

f:id:yiskw713:20210220132957p:plain

参考