Python 에서 Matplotlib 서브플롯 생성하기

소개

Matplotlib 는 선 그래프, 산점도, 막대 그래프, 서브플롯 등과 같은 시각화를 생성하는 데 사용되는 Python 라이브러리입니다. 이 랩에서는 .pyplot.subplot을 사용하여 두 개의 서브플롯이 있는 그림을 만드는 방법을 배우게 됩니다.

VM 팁

VM 시작이 완료되면 왼쪽 상단을 클릭하여 Notebook 탭으로 전환하여 실습을 위해 Jupyter Notebook에 액세스하십시오.

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

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

필요한 라이브러리 가져오기

이 튜토리얼에서는 Matplotlib 라이브러리의 pyplot 모듈과 numpy 라이브러리를 사용합니다.

import matplotlib.pyplot as plt
import numpy as np

데이터 정의

서브플롯을 생성하는 데 사용할 두 개의 데이터 세트를 정의합니다.

def f(t):
    return np.exp(-t) * np.cos(2*np.pi*t)

t1 = np.arange(0.0, 5.0, 0.1)
t2 = np.arange(0.0, 5.0, 0.02)

서브플롯 생성

.pyplot.subplot을 사용하여 두 개의 서브플롯이 있는 figure 를 생성합니다.

plt.figure()

plt.subplot(211)
plt.plot(t1, f(t1), color='tab:blue', marker='o')
plt.plot(t2, f(t2), color='black')

plt.subplot(212)
plt.plot(t2, np.cos(2*np.pi*t2), color='tab:orange', linestyle='--')

plt.show()

subplot() 함수는 세 개의 인수를 받습니다: 행의 수, 열의 수, 그리고 현재 플롯의 인덱스입니다. 인덱스는 왼쪽 상단 모서리에서 1 부터 시작하여 행별로 증가합니다. 이 예제에서는 상단과 하단에 각각 하나씩 두 개의 서브플롯이 있는 figure 를 생성합니다.

첫 번째 서브플롯에서는 t1을 f(t1)에 대해, t2를 f(t2)에 대해 플롯합니다. 첫 번째 플롯의 색상을 파란색으로 설정하고 각 데이터 포인트에 원형 마커를 추가합니다. 두 번째 플롯의 색상은 검정색으로 설정합니다.

두 번째 서브플롯에서는 t2를 2*np.pi*t2의 코사인 함수에 대해 플롯합니다. 플롯의 색상은 주황색으로, 선 스타일은 점선으로 설정합니다.

참조

이 예제에서는 다음 함수, 메서드, 클래스 및 모듈의 사용법을 보여줍니다.

• matplotlib.pyplot.figure
• matplotlib.pyplot.subplot

요약

이 랩에서는 .pyplot.subplot을 사용하여 두 개의 서브플롯이 있는 figure 를 생성하는 방법을 배웠습니다. 두 개의 데이터 세트를 정의하고 서로 다른 색상과 선 스타일을 사용하여 두 개의 서브플롯에 플롯했습니다. 또한 이 예제에서 사용된 함수와 모듈에 대한 참조를 제공했습니다.