Matplot 라이브러리로 간단한 그래프 그리기

동생이 과제로 파이썬을 한다고 하길래 뭔가 하고 봤더니 그래프를 그리는 과제였다.

파이썬으로 그래프 그리는 건 개발 맨~~ 처음에 배울 때 잠깐 발에 물 닿을 정도로만 해 본 적이 있어서 내가 도움을 줄 수 있을 것 같았다.

 

Matplot은 txt 파일의 데이터를 간단하게 그래프로 나타낼 수 있다.

엄청 짧게 코드를 짤 것 같아서 colab을 사용했다.

 

from google.colab import drive
drive.mount('/content/drive')

txt 파일을 굳이 로컬에 저장 안해도 구글드라이브에 마운트 하면 다 갖다 쓸 수 있음 ㅋ

 

import matplotlib.pyplot as plt
import numpy as np

ecg = np.loadtxt("/content/drive/MyDrive/ecg.txt",delimiter=" ",unpack = False)
print(ecg.shape)

# x축
time = ecg[:, 0]
# y축
amplitude = ecg[:, 1]

# plot 세팅
plt.figure(num=1, dpi=100, facecolor='white')
plt.plot(time, amplitude, color="blue", linewidth=0.5)

plt.title('ecg')
plt.xlabel('time(sec)')
plt.ylabel('amplitude')
plt.xlim( 0, 3)
plt.ylim(-1, 1)

plt.show()

numpy로 txt파일을 가져오고, 구분자는 띄어쓰기로 해서 데이터를 구분했다.

데이터들이 이런 식으로 정렬이 되어 있었기 때무네 ㅋ

첫째 행이 x축(time)에 와야 할 값, 둘째 행이 y축(amplitude)에 와야 할 값이므로 0부터 시작해서 끊어주었다.

 

그 다음 나타낼 그래프를 Matplot으로 세팅하는데 사실 대충 구글링 해서 색 같은거 붙여 넣음.

x축, y축 최소값, 최대값 정도만 설정해 주었고

코드를 실행하면 다음과 같다.

이게 바로 심전도 그래프라고 한다 ㅋ

아주아주 간단하게 그래프가 만들어졌다.

 

wikidocs.net/92071

위키독스

위 링크를 참조하면 Matplot에 대한 자세한 설명을 볼 수 있다.

 

과제 끝.