기본적인 pyplot 함수의 사용법을 잘 몰라서 이번에 내용을 정리하려고 한다. Pyplot은 그래프의 시각화를 도와준다. MATLAB과 비슷하게 명령어 스타일로 동작하는 함수의 모음이다. Pyplot 모듈의 각각의 함수를 사용해서 그래프를 만들고 변화를 줄 수 있다. 예를 들어 그래프 영역을 만들고, 몇 개의 선을 표현하고, 레이블로 꾸미는 등의 일을 할 수 있다.
기본 그래프
from matplotlib import pyplot as plt
plt.plot([1, 2, 3, 4])
plt.ylabel('y-label')
plt.show()plt.plot() 함수에 하나의 숫자 리스트를 입력함으로써 아래와 같은 그래프가 그려진다. Matplotlib은 리스트의 값들이 y 값들이라고 가정하고, x값 [0, 1, 2, 3]을 자동으로 만들어낸다.
plt.plot([1, 2, 3, 4], [1, 4, 9, 16])
plt.show()이와 같이 지정하면 각각 x, y값을 지정해서 plotting할 수 있다.
스타일 지정하기
x, y 값 인자에 대해 선의 색상과 형태를 지정하는 포맷 문자열을 세번째 인자에 입력할 수 있다.
from matplotlib import pyplot as plt
plt.plot([1, 2, 3, 4], [1, 4, 9, 16], 'ro')
plt.axis([0, 6, 0, 20])
plt.show()기본 포맷 문자열은 'b-'인데 파란색('blue')의 선('-')을 의미한다.
아래 'ro'는 빨간색('red')의 원형 ('o')마커를 의미합니다.
또한 axis()를 이용해서 축의 범위를 설정줬다. [xmin, xmax, ymin, ymax]
여러개의 그래프 그리기
matplotlib에서는 일반적으로 numpy array를 사용하게 되는데, 사실 numpy array를 사용하지 않더라도 모든 시퀀스는 내부적으로 numpy array로 변환된다.
from matplotlib import pyplot as plt
import numpy as np
# 200ms 간격으로 균일하게 샘플된 시간
t = np.arange(0., 5., 0.2)
# 빨간 dash, 파란 사각형, 녹색 삼각형
plt.plot(t, t, 'r--', t, t**2, 'bs', t, t**3, 'g^')
plt.show()
Grind