기본 개념

대부분의 파이썬 시각화는 matplotlib 라이브러리를 기본으로 합니다. matplotlib.pyplot을 임포트합니다.

import matplotlib.pyplot as plt

그림 그릴 자리 만들기

subplots 함수로 그래프를 그릴 자리를 만듭니다.

fig, ax = plt.subplots()

fig는 그림을, ax는 그림에 그려질 그래프를 말합니다.

만약 2개의 그래프를 하나의 그래프에 그리려면 아래와 같이 합니다.

fig, ax = plt.subplots(ncols=2)

여러 개의 그래프들을 세로로 그리려면 아래처럼 합니다.

fig, ax = plt.subplots(nrows=2)

다음과 같이 하면 하나의 그림에 그래프를 2행 3열로 그립니다.

fix, ax = plt.subplots(nrows=2, ncols=3)

ax.plot을 이용해서 간단한 그래프를 그려보겠습니다. 다음은 (0, 2), (1, 3), 그리고 (2, -1)을 이은 그래프입니다.

fig, ax = plt.subplots()

x = [0, 1, 2]
y = [2, 3, -1]
ax.plot(x, y)

[<matplotlib.lines.Line2D at 0x1d5f7d1e668>]

2개의 그래프를 나란히 그리려면 다음과 같이 하면 됩니다. 왼쪽 ax[0]은 (0, 2)와 (1, 3)을 이은 선이고, 오른쪽 ax[1]은 (4, 1)과 (3, 2)를 이은 선입니다.

fig, ax = plt.subplots(ncols=2)

ax[0].plot([0, 1], [2, 3])
ax[1].plot([4, 3], [1, 2])

[<matplotlib.lines.Line2D at 0x1d5f7e6b5c0>]

그래프의 제목, 가로축 레이블, 세로축 레이블은 아래와 같이 붙입니다. 한글을 사용하려면 한글 글꼴을 설정해주셔야 합니다. 글꼴 설정 방법은 글꼴 설정을 참고하세요.

fig, ax = plt.subplots()

ax.set_title('title')  # 그래프 제목
ax.set_xlabel('x')     # 가로축 레이블
ax.set_ylabel('y')     # 세로축 레이블

Text(0, 0.5, 'y')

그림은 fig.savefig로 저장할 수 있습니다. png, jpg 등의 포맷을 지원합니다. 그래프는 png 포맷으로 저장하는 것이 좋습니다.

fig.savefig('그림.png')