[DL] Matplotlib 라이브러리에 데하여 알아보기

What is Matplotlib?

Matplotlib은 딥러닝 실험을 할때 그래프 그리기 및 데이터를 시각화 해주는 라이브러리입니다.

• Matplotlib을 사용하면 그래프 그리기, 데이터 시각화가 쉬워지고, 주로 데이터 분석을 할때 사용합니다.
• 대표적으로는 선, 막대 그래프, 산점도, 히스토그램 등을 만들수 있습니다.

---

Matplotlib을 이용하여 그래프 그려보기

그래프를 그릴려면 Matplotlib 라이브러리의 Pyplot 모듈을 이용합니다. 한번 sin(사인) 함수를 그리는 예를 한번 살펴 보겠습니다.

import numpy as np
import matplotlib.pyplot as plt

# 데이터 준비
x = np.arange(0, 6, 0.1) # 0부터 6까지 0.1 간격으로 생성
y = np.sin(x)

# 그래프 그리기
plt.plot(x, y)
plt.show()

• 위의 코드를 보면 Numpy의 arange Method를 사용하여 데이터를 생성하여 변수 x에 할당했습니다.
• 변수 x의 각 원소의 Numpy의 sin(사인) 함수인 np.sin()을 적용한 뒤, 변수 y에 할당합니다.
• 그리고 변수 x,y를 인수로 plt.plot Method를 호출해서 그래프를 그리고, plt.show()를 호출해 그래프를 화면에 출력하고 끝납니다.

Sin 함수 그래프

---

Pyplot의 기능

Cos(코사인)함수도 한번 그려보겠습니다. 이번에는 그래프의 제목, 각 label표시 등 pyplot의 다른 기능도 사용해 보겠습니다.

import numpy as np
import matplotlib.pyplot as plt

# 데이터 준비
x = np.arange(0, 6, 0.1) # 0부터 6까지 0.1 간격으로 생성
y1 = np.sin(x)
y2 = np.cos(x)

# 그래프 그리기
plt.plot(x, y1, label="sin")
plt.plot(x, y2, linestyle="--", label="cos") #cos 함수는 점선으로 그리기
plt.xlabel("x")
plt.ylabel("y")
plt.title('sin & cos') # 제목
plt.legend()
plt.show()

sin & cos 함수 그래프

---

이미지 표시하기

pyplot에는 이미지를 표시해주는 Method인 imshow()라고 있습니다. 이미지를 읽어들일때는 matplotlib.image 모듈의 imread() Method를 이용합니다.

import matplotlib.pyplot as plt
from matplotlib.image import imread

img = imread('') # 여기에 이미지의 적절한 경로를 설정하셔야합니다.

plt.imshow(img)
plt.show()

• 위의 코드를 실행하면 읽은 이미지가 표시됩니다.
• 주의하셔야 할점은 사용자의 환경에 맞게 파일 이름, 경로를 적절히 수정해야 합니다.