Để vẽ đường trung bình cho một biểu đồ trong matplotlib, chúng ta có thể thực hiện các bước sau -
-
Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô con.
-
Tạo điểm dữ liệu x và y bằng cách sử dụng numpy.
-
Sử dụng phương thức subplots () để tạo một hình và một tập hợp các vùng phụ.
-
Sử dụng plot () phương pháp cho các điểm dữ liệu x và y.
-
Tìm giá trị trung bình của mảng, x.
-
Lập đồ thị các điểm dữ liệu x và y_avg bằng cách sử dụng plot () phương pháp.
-
Đặt chú giải trên hình vẽ.
-
Để hiển thị hình này, hãy sử dụng show () phương pháp.
Ví dụ
import numpy as np from matplotlib import pyplot as plt plt.rcParams["figure.figsize"] = [7.50, 3.50] plt.rcParams["figure.autolayout"] = True x = np.array([3, 4, 5, 6, 7, 8, 9]) y = np.array([6, 5, 4, 3, 2, 1, 6]) fig, ax = plt.subplots() ax.plot(x, y, 'o-', label='line plot') y_avg = [np.mean(x)] * len(x) ax.plot(x, y_avg, color='red', lw=6, ls='--', label="average plot") plt.legend(loc=0) plt.show()
Đầu ra
