Matplotlib line plot 기본 예제

matplotlib은 데이터를 그래프로 표현하는 대표적인 시각화 라이브러리입니다.

데이터의 시각화는 그 특성을 파악하는데 필수적이기 때문에 데이터 사이언스에서 자주 사용됩니다.

plot이 많이 사용되는데 이것은 데이터를 가상의 공간에 미리 그려놓는 것을 의미합니다.

주로 pyplot모듈을 통해 시각화합니다.

때문에 아래와 같이 import 해줍니다.

 

from matplotlib import pyplot as plt
import matplotlib.pyplot as plt

 

라인, 막대그래프, boxplot, pie, scatter(산포도), 히스토그램 등의 다양한 그래프를 그릴 수 있습니다.

그리기 전에 공통으로 쓰이는 기본 함수에 대해 알아보도록 하겠습니다.

 

plt.plot(리스트 or 판다스 시리즈)
입력 데이터로 plot을 생성합니다.
x,y데이터를 입력해야 합니다.

plt.xlabel(문자열)
plt.ylabel(문자열)
x,y축의 이름을 정할 수 있습니다.

plt.title(문자열)
제목을 정할 수 있습니다.

plt.show()
생성된 그래프를 출력합니다.

plt.savefig()
이미지로 저장할 수 있습니다.

예제를 통해 한 번 사용해 보도록 하겠습니다.

 

from matplotlib import pyplot as plt
x = [1, 2, 3, 4, 5, 6]
y = [1, 2, 3, 4, 5, 6]
plt.figure()
plt.plot( x, y,'--r')
#--r과 같이 선의 모양과 색을 줄 수 있다.

plt.xlabel("x")
plt.ylabel("y")
plt.title("x&y")
plt.xlim(1,10)
plt.ylim(1,10)
#그래프의 구간을 조절하는 함수

plt.show()

from matplotlib import pyplot as plt
x = [1, 2, 3, 4, 5, 6]
y_1 = [1, 2, 3, 4, 5, 6]
y_2 = [10,20,30,40,50,60]
plt.figure()
plt.plot(x, y_1,'r',label='y_1')
plt.plot(x,y_2,'b',label='y_2')
#두가지 데이터를 한 그래프에 표현할 수 있습니다.
#또한 각각의 데이터 마다 label을 붙여줄 수 있습니다.

plt.xlabel("x")
plt.ylabel("y")
plt.title("x&y")

plt.legend(loc="lower right")
#라벨을 표시해 줄 수 있습니다.
plt.show()

 

matplotlob 페이지에 가면 더 많은 기능들을 살펴볼 수 있으니 방문해 보세요 ㅎㅎ

matplotlib.org/gallery/index.html

Gallery — Matplotlib 3.3.2 documentation