파이썬을 이용한 그래프 그리기

이 글에서는 파이썬의 그래프 라이브러리인 matplotlib을 이용해 기본적인 그래프를 그려보려고 해요.

그래프 그리기는 크게 [도화지 그리기 -> 그래프 구역을 정하기 -> 원하는 수식이 담긴 그래프 넣기] 의 순서로 이루어지는데요. 여기에서 그래프를 그리는 방법을 익힌 후 실질적으로 데이터를 적절한 그래프 모형으로 표현하면 좋을 것 같아요.

 

1. plt.figure() : 도화지 그리기

괄호를 비워도 기본값 사이즈로 동작하며 figsize 속성을 추가해서 도화지 사이즈를 설정할 수 있어요.

fig = plt.figure(figsize=(10, 6))

plt.show()

 

 

2. plt.subplot() : 도화지를 일정 구역으로 나누어 그래프 그리기

괄호 안에 행의 수, 열의 수, n번째 수를 입력해 행렬 모양으로 그래프 구역을 만들 수 있어요.

fig = plt.figure()
ax = plt.subplot(2, 2, 1) # 2x2행렬 중 첫번째 구역에 그래프 생성
ax.plot([1, 2, 3, 4])
plt.show()

fig = plt.figure()
ax = plt.subplot(221) # 2,2,1과 같은 의미
ax.plot([1, 2, 3, 4])
plt.show()

 

 

3. fig.add_subplot() : 도화지를 일정 구역으로 나누어 그래프 그리기

plt.subplot()과 마찬가지로 지정한 행렬 중 하나의 구역에 그래프를 만들 수 있어요. 차이점은 subplot()은 이전의 그래프 행렬을 지우고 새로 지정하지만 add_subplot()은 지우지 않고 그 위에 그래프를 그립니다. 

fig = plt.figure()
ax1 = fig.subplot(333)   # 그래프를 그린 다음에 다른 그래프를 그려도 사라지지 않음
ax2 = fig.add_subplot(2, 2, 1)
ax2.plot([1, 2, 3, 4])
plt.show()

 

4.plt.subplots : 도화지 생성과 그래프 행렬 동시에 생성

기존에는 plt.figure()로 도화지를 만들고 add_subplot이나 subplot을 이용해 그래프를 하나씩 생성했다면, 여기에서는 그래프를 한꺼번에 지정한 행렬만큼 만들 수 있어요.

 

fig, axes = plt.subplots(2, 2)
((ax1, ax2), (ax3, ax4)) = axes #axes가 시리즈 형식으로 생성되므로 각 그래프 구역별 변수를 지정하기
#ax[0][1].plot([1, 2, 3, 4])  이런 방식으로도 가능!

ax1.plot([1, 2, 3, 4])

plt.show()