Bagaimana cara menambahkan garis rata-rata ke plot di matplotlib

Anda dapat menggunakan sintaks dasar berikut untuk menambahkan garis rata-rata ke plot di 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 ))

Perhatikan bahwa axhline menambahkan garis horizontal ke plot dan nanmean menghitung nilai rata-rata (mengabaikan NaN) di mana garis harus ditempatkan.

Contoh berikut menunjukkan cara menggunakan sintaksis ini dalam praktiknya.

Contoh: Tambahkan garis rata-rata ke plot di Matplotlib

Misalkan kita memiliki panda DataFrame berikut:

 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

Kita dapat menggunakan kode berikut untuk membuat plot sebar x versus y dan menambahkan garis horizontal yang mewakili nilai rata-rata 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 )) 

Kita dapat melihat bahwa garis rata-rata telah ditambahkan ke plot tepat di atas nilai y sebesar 8.

Jika kita menghitung nilai rata-rata y, kita mendapatkan hasilnya adalah 8,333:

 #calculate average y-value
n.p. nanmean (df. y )

8.333333333

Perhatikan bahwa kita juga dapat menggunakan argumen color , linestyle dan linewidth untuk menentukan masing-masing warna, linetype dan linewidth dari garis rata-rata:

 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 ') 

Catatan : Anda dapat menemukan dokumentasi online lengkap untuk fungsi axhline() di sini .

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya di Matplotlib:

Bagaimana cara menambahkan garis tren di Matplotlib
Cara menggambar garis vertikal di Matplotlib
Cara memplot banyak baris di Matplotlib