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

Matplotlib로 Seaborn 플롯 범례(legend) 글꼴 크기 키우는 방법

Seaborn 플롯에서 범례(legend)의 글꼴 크기를 키우려면 fontsize 변수를 정의한 뒤, Matplotlib의 legend() 메서드 인자로 전달하면 됩니다. 이 방법은 별도의 복잡한 설정 없이 한 줄의 코드로 범례 텍스트 크기를 자유롭게 조절할 수 있어 매우 간편합니다.

구현 단계

  • Pandas를 사용해 데이터프레임을 생성합니다. 여기서는 number, count, select 세 개의 키(열)를 사용합니다.

  • Seaborn의 barplot() 메서드로 막대그래프를 그립니다.

  • 범례의 글꼴 크기를 지정할 변수 fontsize를 초기화합니다.

  • legend() 메서드에 fontsize 값을 인자로 전달하여 그림 위에 범례를 배치합니다.

  • show() 메서드를 호출해 최종 그림을 화면에 출력합니다.

예제 코드

import pandas
import matplotlib.pylab as plt
import seaborn as sns
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
df = pandas.DataFrame(dict(
    number=[2, 5, 1, 6, 3],
    count=[56, 21, 34, 36, 12],
    select=[29, 13, 17, 21, 8]
))
bar_plot1 = sns.barplot(x='number', y='count', data=df, label="count", color="red")
bar_plot2 = sns.barplot(x='number', y='select', data=df, label="select", color="green")
fontsize = 20
plt.legend(loc="upper right", frameon=True, fontsize=fontsize)
plt.show()

코드 설명

위 예제에서는 먼저 rcParams 설정을 통해 그림 크기(7.00 × 3.50)와 자동 레이아웃 조정을 활성화했습니다. 그다음 두 개의 막대그래프를 겹쳐 그리는데, 하나는 빨간색으로 'count' 데이터를, 다른 하나는 초록색으로 'select' 데이터를 나타냅니다.

핵심 부분은 fontsize = 20으로 범례 글꼴 크기를 지정하고, plt.legend(loc="upper right", frameon=True, fontsize=fontsize)를 통해 오른쪽 상단에 테두리가 있는 범례를 배치하는 것입니다. fontsize 값을 원하는 숫자로 변경하면 범례 텍스트 크기가 그에 맞게 조절됩니다.

실행 결과

코드를 실행하면 두 개의 막대그래프가 함께 표시되고, 오른쪽 상단에 글꼴 크기 20pt로 확대된 범례('count'와 'select')가 나타납니다.