[밑러닝] 1.6 matplotlib, 파이썬의 시각화 툴

* 이 글은 <밑바닥부터 시작하는 딥러닝2 (저자: 사이토 고키)> 책을 읽으며 정리한 글입니다.
* 나중에라도 제가 참고하기 위해 정리해 두었으며, 모든 내용을 적은 것이 아닌,
필요하다고 생각되는 부분만 추려서 정리한 것임을 미리 밝힙니다.

---

목차

1. matplotlib의 정의
2. 그래프 그려보기
3. 사진 재현

---

1. matplotlib의 정의

matplotlib은 파이썬에서 쓰이는 시각화 툴이다. 그래프 그리기에 탁월하다.
(이외에도 seaborn이라는 툴이 있는데 이는 빅데이터를 시각화하기에 적당한 툴이다)
일단은 matplotlib에 대해 서술해보도록 하자.
matplotlib은 외부 라이브러리고, 흔히 plt로 줄여서 사용한다.

import matplotlib.pyplot as plt

 

2. 그래프 그려보기

앞서 배운 넘파이(numpy)를 이용해 간단한 그래프를 그려보자. 코드는 밑바닥부터 시작하는 딥러닝 책 42페이지를 참고했다.

import numpy as np
import matplotlib.pyplot as plt

# 데이터 준비
x = np.arange(0, 6, 0.1) # 0에서 6까지 0.1 간격으로 설정
y = np.sin(x)

# 그래프 그리기
plt.plot(x, y)
plt.show()

plt.xlabel('x축 이름')을 통해 x축의 이름을 지정할 수 있고, plt.ylabel('y축 이름')을 통해 y축의 이름을 지정할 수도 있다.
또한 plt.title('그래프의 제목')으로 그래프의 제목을 지정할 수 있으며, plt.legend()을 통해 범례를 나타나게 할 수 있다. 이 모든 것을 코드로 적용하면 다음과 같다.

import numpy as np
import matplotlib.pyplot as plt

# 데이터 준비
x = np.arange(0, 6, 0.1) # 0에서 6까지 0.1 간격으로 설정
y = np.sin(x)

# 그래프 그리기
plt.plot(x, y, label ='sin')
plt.xlabel('x') # x축 이름 지정
plt.ylabel('y') # y축 이름 지정
plt.title('sin graph') # 그래프의 제목 지정
plt.legend() # 범례 표시
plt.show()

 

3. 사진 재현

python에서 이미지를 불러들이고 재현하는 매서드가 있다. imshow()인데, 다음과 같이 사용한다.

import matplotlib.pyplot as plt
from matplotlib.image import imread

img = imread('test.png') # 괄호 안에 들어가는 것은 파일의 경로를 지정해줘야 하며, 
                         # 지금 같은 경우에는 test.png가 파이썬의 실행 파일 경로에 동일하게 존재하는 경우이다.

plt.imshow(img)
plt.show()

 

---

정리

matplotlib.pyplot에는 일반적인 선 그래프 말고도, 막대 그래프 등 여러가지 형태의 그래프가 있다. 이에 대해서 나중에 자세히 후술하도록 하겠다.