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 で複数の線をプロットする方法