Cara menambahkan garis tren di matplotlib (dengan contoh)

Anda dapat menggunakan sintaks dasar berikut untuk menambahkan garis tren ke plot di Matplotlib:

 #create scatterplot
plt. scatter (x,y)

#calculate equation for trendline
z = np. polyfit (x, y, 1)
p = np. poly1d (z)

#add trendline to plot
plt. plot (x, p(x))

Contoh berikut menunjukkan cara menggunakan sintaksis ini dalam praktiknya.

Contoh 1: Buat garis tren linier di Matplotlib

Kode berikut menunjukkan cara membuat garis tren dasar untuk plot sebar di Matplotlib:

 import numpy as np
import matplotlib. pyplot as plt

#define data
x = np. array ([8, 13, 14, 15, 15, 20, 25, 30, 38, 40])
y = np. array ([5, 4, 18, 14, 20, 24, 28, 33, 30, 37])

#create scatterplot
plt. scatter (x,y)

#calculate equation for trendline
z = np. polyfit (x, y, 1 )
p = np. poly1d (z)

#add trendline to plot
plt. plot (x, p(x)) 

Titik biru melambangkan titik data dan garis biru lurus melambangkan garis tren linier.

Perhatikan bahwa Anda juga dapat menggunakan argumen color , linewidth dan linestyle untuk mengubah tampilan garis tren:

 #add custom trendline to plot
plt. plot (x, p(x), color=" purple ", linewidth= 3 , linestyle=" -- ")

Contoh 2: Buat Garis Tren Polinomial di Matplotlib

Untuk membuat garis tren polinomial, cukup ubah nilai dalam fungsi np.polyfit() .

Misalnya, kita dapat menggunakan nilai 2 untuk membuat garis tren kuadrat:

 import numpy as np
import matplotlib. pyplot as plt

#define data
x = np. array ([8, 13, 14, 15, 15, 20, 25, 30, 38, 40])
y = np. array ([5, 4, 18, 14, 20, 24, 28, 33, 30, 37])

#create scatterplot
plt. scatter (x,y)

#calculate equation for quadratic trendline
z = np. polyfit (x,y, 2 )
p = np. poly1d (z)

#add trendline to plot
plt. plot (x, p(x)) 

Perhatikan bahwa garis tren sekarang melengkung, bukan lurus.

Garis tren polinomial ini sangat berguna ketika data Anda memiliki pola non-linier dan garis lurus gagal menangkap tren data secara memadai.

Sumber daya tambahan

Tutorial berikut menjelaskan cara menjalankan fungsi umum lainnya di Matplotlib:

Cara menyembunyikan sumbu di Matplotlib
Cara memutar label tanda centang di Matplotlib
Bagaimana cara mengubah jumlah centang di Matplotlib