Matplotlib 플롯 범례 생성

소개

Matplotlib 는 사용자가 다양한 유형의 그래프와 플롯을 생성할 수 있도록 해주는 Python 라이브러리입니다. 이 튜토리얼에서는 Matplotlib 플롯에서 범례를 만드는 과정을 안내합니다.

VM 팁

VM 시작이 완료되면, 왼쪽 상단 모서리를 클릭하여 Notebook 탭으로 전환하여 실습을 위해 Jupyter Notebook에 접근하십시오.

때로는 Jupyter Notebook 이 로딩을 완료하는 데 몇 초 정도 기다려야 할 수 있습니다. Jupyter Notebook 의 제한으로 인해 작업의 유효성 검사는 자동화될 수 없습니다.

학습 중에 문제가 발생하면 언제든지 Labby 에게 문의하십시오. 세션 후 피드백을 제공해주시면 문제를 신속하게 해결해 드리겠습니다.

필요한 라이브러리 임포트

시작하기 전에 필요한 라이브러리를 임포트해야 합니다. 이 경우 Matplotlib 라이브러리를 사용합니다.

import matplotlib.pyplot as plt

figure 와 subplot 생성

데이터를 플롯하기 위해 figure 와 subplot 을 생성해야 합니다. 두 개의 subplot 이 있는 플롯을 생성할 것입니다.

fig = plt.figure()

ax = fig.add_subplot(211)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

ax = fig.add_subplot(223)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

플롯에 범례 추가

이제 플롯에 범례를 추가하겠습니다. Matplotlib 에서 범례를 추가하는 두 가지 방법이 있습니다. 이 예제에서는 두 가지 방법을 모두 사용합니다.

## Method 1: Place a legend above the subplot
ax.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc='lower left',
           ncols=2, mode="expand", borderaxespad=0.)

## Method 2: Place a legend to the right of the subplot
ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left', borderaxespad=0.)

플롯 표시

마지막으로, 플롯을 표시합니다.

plt.show()

요약

이 튜토리얼에서는 Matplotlib 플롯에 범례를 추가하는 방법을 배웠습니다. 범례를 추가하기 위해 두 가지 다른 방법을 사용했습니다. 첫 번째 방법은 범례를 서브플롯 위에 배치하고, 두 번째 방법은 범례를 서브플롯 오른쪽에 배치했습니다.