Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách trực quan hóa mối quan hệ tuyến tính bằng Seaborn trong Python

Seaborn cung cấp regplot (cấp thấp) và lmplot (cấp cao, hỗ trợ FacetGrid) để vẽ scatter plot kèm đường hồi quy tuyến tính.

Ví dụ với dataset Tips

import seaborn as sb
from matplotlib import pyplot as plt

my_df = sb.load_dataset('tips')

# regplot: vẽ trực tiếp lên axes hiện tại
sb.regplot(x="total_bill", y="tip", data=my_df)

# lmplot: tạo FacetGrid, hỗ trợ chia nhóm
sb.lmplot(x="total_bill", y="tip", data=my_df)

plt.show()

Giải thích

  • Cả hai đều vẽ điểm dữ liệu và đường hồi quy tuyến tính (OLS) với khoảng tin cậy 95%.
  • regplot linh hoạt hơn cho tùy chỉnh đơn lẻ.
  • lmplot phù hợp khi cần chia lưới (facet) theo biến phân loại (hue, col, row).

Khi nào dùng cái nào?

  • Vẽ nhanh một biểu đồ → regplot.
  • So sánh nhiều nhóm, tạo lưới biểu đồ → lmplot.