Python 데이터 시각화 기초

matplotlib와 seaborn을 이용한 데이터 시각화 기초를 알아봅시다.

14
수강중

1. 기본 개념

동영상이 없는 텍스트 강의 자료입니다.

대부분의 파이썬 시각화는 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')