여러 서브플롯과 그림

지금까지는 하나의 Figure에 하나의 그래프만 그렸습니다. 그렇다면 여러 그래프를 나란히 비교하고 싶을 때는 어떻게 해야 할까요?

subplots를 사용하면 하나의 Figure 안에 여러 그래프를 배치할 수 있습니다.

plt.subplot(rows, cols, index)를 사용해 그림을 격자로 나눌 수 있습니다.

복사
import matplotlib.pyplot as plt

x = [1, 2, 3, 4]
y1 = [2, 4, 1, 3]
y2 = [3, 1, 5, 2]

plt.subplot(1, 2, 1)  # 1행 2열 중 첫 번째 그래프
plt.plot(x, y1)
plt.title("그래프 A")

plt.subplot(1, 2, 2)  # 1행 2열 중 두 번째 그래프
plt.plot(x, y2)
plt.title("그래프 B")

plt.tight_layout()
plt.show()

행과 열의 개수를 지정해 전체 배치를 정하고, 인덱스(index) 값을 지정해 위치를 선택합니다.

새 Figure를 시작하려면 plt.figure()를 사용합니다. 각 Figure은 서로 독립적입니다.

복사
plt.figure(1)
plt.plot([1, 2, 3], [4, 5, 6])
plt.title("그림 1")

plt.figure(2)
plt.plot([1, 2, 3], [6, 5, 4])
plt.title("그림 2")

plt.show()

이러한 기능은 여러 개의 서로 다른 그래프를 그릴 때 유용합니다.

• 같은 페이지에서 관련 데이터를 비교할 때: 서브플롯 사용
• 서로 다른 맥락의 시각화를 만들 때: 여러 Figure 사용