[python] 그래프 데이터를 변수로 관리하기

출처

matplotlib을 이용하여 더 그래프를 잘 만드려면 변수로 관리하는 방법을 알아야 합니다.
matplotlib.pyplot에 직접 접근해서 그래프를 생성하는 방법으로는 여러 개의 그래프를 배치하고 관리하기엔 한계가 있습니다.
그래프 데이터를 객체(Object)로 이해하고 변수에 저장하여 관리해보도록 합시다.

# 필요한 모듈을 참조합니다.
from matplotlib import pyplot as plt
import numpy as np
from math import pi

# 그래프 그릴 값들을 생성합니다.
xs = np.arange(-1*pi, 1*pi, 0.05)
ys_sin = np.sin(xs) # xs에 sin()함수를 적용한 y들의 값을 생성

# 변수에 빈 캔버스를 제공하는 Figure(그림) 인스턴스(데이터)를 생성하고 할당합니다.
fig = plt.figure() 

# Figure 인스턴스에 좌표축 데이터(axes)를 설정합니다.
# 좌표축 데이터만 따로 쉽게 관리하게 변수 ax로 저장해둡시다.
# add_axes()에는 위치(좌우, 위아래), 너비, 높이의 값. 총 4개의 값을 0~1로 넣어주어야 합니다.
ax = fig.add_axes([0,0,1,1])
ax.plot(xs, ys_sin)

# 여기에 필요한 제목 등을 설정합니다.
ax.set_title('sin function wave')
ax.set_xlabel('angle')
ax.set_ylabel('value')

# 격자도 추가합시다.
ax.grid()  # 격자 추가

# 화면을 봅니다.
# 원래라면 plt.show(), colab에서는 fig
# colab에서는 figure 생성부터 ax.plot() 코드까지 있으면 자동으로 그래프가 나옵니다.
plt.show()