Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Seaborn 선형 회귀 조인트플롯에서 회귀선 색상 변경하는 방법

Seaborn의 선형 회귀 조인트플롯(jointplot)에서 회귀선의 색상을 변경하려면 jointplot() 메서드에 joint_kws 매개변수를 사용하면 됩니다. 이 매개변수는 산점도와 회귀선에 적용되는 스타일 옵션을 딕셔너리 형태로 전달할 수 있게 해줍니다.

구현 단계

  • 그림(figure) 크기를 설정하고 서브플롯 사이 및 주변 여백을 자동으로 조정합니다.
  • NumPy를 사용해 x, y 데이터 포인트를 생성한 뒤, 이를 바탕으로 Pandas 데이터프레임을 만듭니다.
  • jointplot() 메서드를 호출할 때 인자에 joint_kws를 포함하여 원하는 선 색상을 지정합니다.
  • show() 메서드를 사용해 최종 그림을 화면에 표시합니다.

예제 코드

import seaborn as sns
import numpy as np
from matplotlib import pyplot as plt
import pandas as pd

# 그림 크기 및 레이아웃 설정
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

# 난수 기반으로 x, y 데이터 생성
X = np.random.randn(1000,)
Y = 0.2 * np.random.randn(1000) + 0.5

# 데이터프레임 생성 후 조인트플롯 그리기
df = pd.DataFrame(dict(x=X, y=Y))
g = sns.jointplot(x="x", y="y", data=df, kind='reg', height=3.5, joint_kws={'color':'green'})

plt.show()

실행 결과

위 코드를 실행하면 중앙에는 초록색(green) 회귀선이 표시된 산점도가 나타나고, 상단과 우측에는 각 변수의 분포를 보여주는 히스토그램이 함께 출력됩니다. joint_kws의 'color' 값을 'red', 'blue' 등 다른 색상으로 바꾸면 회귀선 색상도 손쉽게 변경할 수 있습니다.

Seaborn 선형 회귀 조인트플롯에서 회귀선 색상 변경하는 방법