Seaborn regplot에서 점과 회귀선에 서로 다른 색상 적용하는 방법

Seaborn의 regplot()은 산점도와 회귀선을 함께 시각화해 주는 강력한 함수입니다. 기본 설정에서는 점과 회귀선이 동일한 색상으로 표시되지만, scatter_kws와 line_kws 매개변수를 활용하면 두 요소에 각각 다른 색상을 지정할 수 있습니다.

구현 단계

• 그림(figure) 크기를 설정하고 서브플롯 주변 및 사이의 여백을 조정합니다.

• X축과 Y축 키를 가진 Pandas 데이터프레임을 생성합니다.

• sns.regplot()을 호출하여 수치형 독립 변수에 대한 회귀 모델을 플롯합니다. 이때 scatter_kws={"color": "red"}로 점의 색상을, line_kws={"color": "green"}으로 회귀선의 색상을 각각 지정합니다.

• plt.show() 메서드를 사용하여 그림을 화면에 표시합니다.

예제 코드

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()

실행 결과

위 코드를 실행하면 산점도의 점들은 빨간색으로, 회귀선은 초록색으로 표시된 그래프를 확인할 수 있습니다.

핵심 포인트 정리

scatter_kws는 산점도 점의 스타일(색상, 크기, 투명도 등)을 제어하는 딕셔너리이며, line_kws는 회귀선의 스타일을 제어합니다. 이 두 매개변수에는 Matplotlib에서 사용하는 키워드 인자를 그대로 전달할 수 있으므로, 색상 외에도 alpha, s(점 크기), linewidth 등 다양한 옵션을 조합해 시각화를 더욱 세밀하게 꾸밀 수 있습니다.