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

Hiển thị màu khác nhau cho điểm và đường hồi quy trong Seaborn regplot

Để tô màu khác nhau cho điểm dữ liệu (scatter) và đường hồi quy (line) trong sns.regplot(), ta dùng các tham số scatter_kws và line_kws.

Các bước

  • Thiết lập figure size và autolayout.
  • Tạo DataFrame với cột X-Axis và Y-Axis.
  • Gọi sns.regplot() với scatter_kws={"color": "red"} và line_kws={"color": "green"}.
  • Hiển thị bằng show().

Ví dụ

import pandas
import matplotlib.pylab as plt
import seaborn as sns
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pandas.DataFrame({"X-Axis": [np.random.randint(5) for i in range(10)], "Y-Axis": [np.random.randint(5) for i in range(10)]})
sns.regplot(x='X-Axis', y='Y-Axis', data=df, scatter_kws={"color": "red"}, line_kws={"color": "green"})
plt.show()

Kết quả

Hiển thị màu khác nhau cho điểm và đường hồi quy trong Seaborn regplot