Để vẽ đường xu hướng phân tán bằng matplotlib, chúng ta có thể sử dụng polyfit () và poly1d () các phương pháp để lấy điểm của đường xu hướng.
Các bước
-
Đặ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 các điểm dữ liệu x và y bằng cách sử dụng numpy.
-
Tạo một hình và một tập hợp các ô phụ.
-
Lập đồ thị các điểm dữ liệu x và y bằng cách sử dụng numpy.
-
Tìm các điểm dữ liệu của đường xu hướng bằng cách sử dụng polyfit () và poly1d () phương pháp.
-
Lô x và p (x) điểm dữ liệu bằng cách sử dụng plot () phương pháp.
-
Để 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.random.rand(100) y = np.random.rand(100) fig, ax = plt.subplots() _ = ax.scatter(x, y, c=x, cmap='plasma') z = np.polyfit(x, y, 1) p = np.poly1d(z) plt.plot(x, p(x), "r-o") plt.show()
Đầu ra