[kr-dev] 파이썬으로 차트 그리기 기초

지난번에 너무 많은 내용을 하나의 포스팅에 넣었더니 평가가 저조하더군요.
https://steemit.com/kr/@minari/kr-dev-cryptocompare

그래서 이번에는 짧게 기초적인 파이썬 그래픽 라이브러리 사용법에 대한 설명을 드리려고 합니다.

파이썬 언어의 편리함에 힘입어 수많은 고수들이 각종 라이브러리를 파이썬 용으로 만들어 배포하고 있으며, 대부분 공짜로 사용할 수 있습니다.

그 중에서도 데이터 분석에 사용되는 pandas와 더불어 많이 사용되는 라이브러리는 matplotlib입니다.

결국 뭔가를 분석하고 나면 그 결과를 확인해야 하는데, 수치보다는 그래프가 더 직관적으로 들어오기 때문입니다.

현재 matplotlib에서 제공하고 있는 차트의 종류는 수백가지가 넘으며, 사실상 현실에서 사용하는 모든 종류의 차트가 라이브러리화되어 함수 하나로 호출해서 사용할 수 있습니다. (예, 선차트 plot(), 바차트 bar() ...)

https://matplotlib.org/

회사에서 보고서로 쓰이는 차트들은 대부분 정형화되어 있어서 엑셀의 기능으로도 충분하지만, 정형화되어 있지 않은 데이터 분석의 결과를 확인하기에는 부족한 면이 없지 않습니다.

그래서 파이썬을 통한 데이터 분석에 앞서서 그래픽 라이브러리를 이용하는데 조금 친숙해지시라는 의미로 간단한 예제를 만들었습니다.

두 개의 직선과 곡선을 하나의 차트에 넣고, 제목과 x,y축 및 범례를 표시해 보겠습니다.

프로그램 코드는 다음과 같으며, 파이썬은 코드 영역 구분을 {}블럭 설정 대신 들여쓰기로 하기 때문에 텍스트를 스팀이세 복사해 넣으면 들여쓰기가 없어지지만, 아래 코드는 애초에 들여쓰기가 없어서, 바로 복사해서 실행해 보실수 있습니다.

import matplotlib.pylab as plt
import numpy as np

# xy평면에 그릴 직선과 곡선에 대한 함수
x = np.arange(0, 12, 0.1)
y = np.sin(x)
y2 = x / 6 - 1

# 곡선과 직선을 matplotlib 라이브러리를 이용해 그림
plt.plot(x, y, 'r--', lw=2, label='sin') # 선 색빨은 빨강, 두께는 2
plt.plot(x, y2, 'b', label='line') # 선 색깔은 파랑

# 차트 꾸미기
plt.xlim(-1, 15) # x축 범위
plt.ylim(-1.1, 1.2) # y축 범위
plt.grid() # 차트에 격자무늬
plt.title("Title") # 차트 제목
plt.xlabel("X axis") # x축 설명
plt.ylabel("Y axis") # y축 설명
plt.legend(loc='best') # 범례 표시 (위치 = 알아서)
plt.show() # 차트 보이기

위와 같은 프로그램을 파이썬을 지원하는 편집기 (spyder, pycharm, jupyter notebook 등)에 입력하여 실행을 시켜보면, 출력창 또는 팝업창으로 아래와 같은 차트를 보실 수 있습니다.

개발자나 이공계가 아니더라도, 파이썬 정도의 언어를 하나 익혀주신다면 본인의 업무능력이나 취미활동 또는 향후 자녀 교육에도 도움이 도움이 되리라 생각됩니다. ^^

---

※ 파이썬을 이용한 투자 포스팅은 #kr-systemtrading에서 보실 수 있습니다.