Hướng dẫn how do you code a line of best fit in python? - làm thế nào để bạn viết một dòng phù hợp nhất trong python?

Bạn có thể sử dụng cú pháp cơ bản sau đây để vẽ một dòng phù hợp nhất với Python:

#find line of best fit
a, b = np.polyfit(x, y, 1)

#add points to plot
plt.scatter(x, y)

#add line of best fit to plot
plt.plot(x, a*x+b)

Ví dụ sau đây cho thấy cách sử dụng cú pháp này trong thực tế.

Ví dụ 1: Lô đường cơ bản phù hợp nhất với Python

Mã sau đây cho thấy cách vẽ một dòng cơ bản phù hợp nhất với Python:

import numpy as np
import matplotlib.pyplot as plt

#define data
x = np.array([1, 2, 3, 4, 5, 6, 7, 8])
y = np.array([2, 5, 6, 7, 9, 12, 16, 19])

#find line of best fit
a, b = np.polyfit(x, y, 1)

#add points to plot
plt.scatter(x, y)

#add line of best fit to plot
plt.plot(x, a*x+b)        

Ví dụ 2: Lô đường tùy chỉnh phù hợp nhất với Python

Mã sau đây cho thấy cách tạo cùng một dòng phù hợp nhất như ví dụ trước ngoại trừ với các bổ sung sau:

• Màu sắc tùy chỉnh cho các điểm và dòng phù hợp nhất
• Kiểu và chiều rộng tùy chỉnh cho dòng phù hợp nhất
• Phương trình của đường hồi quy được trang bị được hiển thị trên lô

import numpy as np
import matplotlib.pyplot as plt

#define data
x = np.array([1, 2, 3, 4, 5, 6, 7, 8])
y = np.array([2, 5, 6, 7, 9, 12, 16, 19])

#find line of best fit
a, b = np.polyfit(x, y, 1)

#add points to plot
plt.scatter(x, y, color='purple')

#add line of best fit to plot
plt.plot(x, a*x+b, color='steelblue', linestyle='--', linewidth=2)

#add fitted regression equation to plot
plt.text(1, 17, 'y = ' + '{:.2f}'.format(b) + ' + {:.2f}'.format(a) + 'x', size=14)

Hãy thoải mái đặt phương trình hồi quy được trang bị trong bất cứ điều gì (x, y) bạn muốn trên cốt truyện.(x, y) coordinates you would like on the plot.

Đối với ví dụ cụ thể này, chúng tôi đã chọn (x, y) = (1, 17).

Tài nguyên bổ sung

Các hướng dẫn sau đây giải thích cách phù hợp với các mô hình hồi quy khác nhau trong Python:

Hướng dẫn đầy đủ về hồi quy tuyến tính trong Python Cách thực hiện hồi quy đa thức trong Python Cách thực hiện hồi quy lượng tử trong Python
How to Perform Polynomial Regression in Python
How to Perform Quantile Regression in Python