[머신러닝] python matplotlib 사용법 - line plot, scatter plot

 

matplotlib

matplotlib은 다양한 데이터를 많은 방법으로 도식화 할 수 있도록 하는 파이썬 라이브러리.

matplotlib를 이용하면 numpy로 만든 자료구조를 쉽게 시각화를 할 수 있습니다.

 

matplotlib를 사용하기 위해서는 우선 matplotlib를 설치하고 아래와 같이 import를 해주어야합니다.

import matplotlib

import matplotlib.pyplot as plt

 

실행을 하기위해서는 코드 제일 위에 #%%를 작성해주어야합니다. (VSC 기준)

 

line plot

#%%
import matplotlib.pyplot as plt

x_data = [ x for x in range(-5,5) ] 
y_data = [ y*y for y in range(-5,5) ] 

plt.title('line plot')
plt.grid() 
plt.plot(x_data, y_data ,color='b') 
plt.show()

x와 y 데이터를 각각 만들고 그래프를 출력하게 되면,

line plot 예제

 

scatter plot

#%%
import matplotlib.pyplot as plt
import numpy as np

# x data,  y data 생성
x_data = np.random.rand(100)
y_data = np.random.rand(100) 

plt.title('scatter plot')
plt.grid() 
plt.scatter(x_data, y_data, color='b', marker='o') 
plt.show()

x와 y 데이터를 각각 0에서 1사이의 난수 100개로 지정하면,

scatter plot 예제

 

이 두가지 방식은 머신러닝에서 자주 사용될 방식입니다.