Matplotlib でプロットに平均線を追加する方法
次の基本構文を使用して、Matplotlib のプロットに平均線を追加できます。
import matplotlib. pyplot as plt import numpy as np #create scatterplot plt. scatter (df. x , df. y ) #add horizontal line at mean value of y plt. axhline (y=np. nanmean (df. y ))
axhline はプロットに水平線を追加し、 nanmean は線を配置する平均値を (NaN を無視して) 計算することに注意してください。
次の例は、この構文を実際に使用する方法を示しています。
例: Matplotlib のプロットに平均線を追加する
次のパンダ データフレームがあるとします。
import pandas as pd
#createDataFrame
df = pd. DataFrame ({' x ': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
' y ':[2, 5, 6, 5, 7, 8, 10, 12, 10, 9, 11, 15]})
#view first five rows of DataFrame
df. head ()
x y
0 1 2
1 2 5
2 3 6
3 4 5
4 5 7
次のコードを使用して、x と y の散布図を作成し、平均 y 値を表す水平線を追加できます。
import matplotlib. pyplot as plt import numpy as np #create scatterplot plt. scatter (df. x , df. y ) #add horizontal line at mean value of y plt. axhline (y=np. nanmean (df. y ))
平均線がプロットの y 値 8 のすぐ上に追加されていることがわかります。
平均 y 値を計算すると、8.333 であることがわかります。
#calculate average y-value n.p. nanmean (df. y ) 8.333333333
color 、 linestyle 、 linewidth引数を使用して、それぞれ平均線の色、線種、線幅を指定することもできることに注意してください。
import matplotlib. pyplot as plt import numpy as np #create scatterplot plt. scatter (df. x , df. y ) #add horizontal line at mean value of y plt. axhline (y=np. nanmean (df. y ), color=' red ', linestyle=' -- ', linewidth= 3 , label=' Avg ')
注: axhline()関数の完全なオンライン ドキュメントは、ここで見つけることができます。
追加リソース
次のチュートリアルでは、Matplotlib で他の一般的なタスクを実行する方法を説明します。
Matplotlib で近似曲線を追加する方法
Matplotlib で垂直線を引く方法
Matplotlib で複数の線をプロットする方法