[Python] 파이썬 그래프 그리기/ matplotlib 패키지 사용법

matplotlib 패키지

파이썬에서 그래프를 그릴 수 있는 matplotlib 패키지를 제공한다.

 

데이터를 시각화하여 보기 위해 사용한다.

 

 

matpltlib import

matpltlib 패키지를 import 해주자.

 

이때 plt라는 이름으로 import 해주는 것이 관례다.

 

matpltlib 패키지가 없다면 설치 하자.

import matplotlib.pyplot as plt

 

 

기본적인 그래프 그려보기

가장 기본적인 그래프다.

 

아래 예제를 실행 해 보면 아래 그림과 같은 결과가 나온다.

import matplotlib.pyplot as plt #그래프 그리는 패키지 import

x = [10,20,30,40,50]
y = [10,20,30,40,50]

plt.plot(x, y, color = 'blue', marker = 'o', alpha = 1, linewidth = 1)
plt.xlabel('y')
plt.ylabel('X')
plt.show()  #그래프 띄어주기

결과 :

 

 

 

 

 

예제 읽어보면 대충 어떤 방식으로 그래프가 만들어지는지 알거다.

 

그럼 그래프를 세팅해주는 plot 함수에 대해 알아보자.

 

 

 

plot 함수

plot 함수는 matpltlib 패키지가 제공하는 다양한 그래프 중 하나이며, 많이 사용된다.

 

plot  함수에 들어가는 매개변수에 대해 알아보자.

plt.plot(x, y, color = 'blue', marker = 'o', alpha = 1, linewidth = 1)

 

 

 

 

다른 그래프도 사용하려면 아래 링크를 참조하면 된다.

https://matplotlib.org/3.5.0/plot_types/index.html

Plot types — Matplotlib 3.5.0 documentation

 

 

 

 

 

x, y

점의 위치다. 예제에서는 리스트로 x, y 위치를 지정했다.

plt.plot(x, y, color = 'blue', marker = 'o', alpha = 1, linewidth = 1)

 

 

color

선의 색상을 정해준다. blue 뿐만 아니라 red, yellow 등 다양한 색상을 적어서 적용 할 수 있다.

또한  #aabbcc 처럼 RGB로 도 선택 할 수 있다.

plt.plot(x, y, color = 'blue', marker = 'o', alpha = 1, linewidth = 1)

 

 

marker

점의 모양을 그려준다. 원하는 모양을 적어주면 된다.

plt.plot(x, y, color = 'blue', marker = 'o', alpha = 1, linewidth = 1)

 

아래 사이트에서 여러가지 모양을 볼 수 있다.

https://matplotlib.org/3.5.0/api/markers_api.html#module-matplotlib.markers

matplotlib.markers — Matplotlib 3.5.0 documentation

 

 

 

 

 

 

alpha

0 ~ 1 까지 선의 투명도를 선택 한다. 낮을 수록 투명 해진다.

plt.plot(x, y, color = 'blue', marker = 'o', alpha = 1, linewidth = 1)

 

 

 

linewidth

선의 굵기를 선택한다. 정수, 소수 둘 다 가능하다.

plt.plot(x, y, color = 'blue', marker = 'o', alpha = 1, linewidth = 1)

 

 

---

 

아무런 세팅 없이 그래프 그리기

 

plot 함수에 아무런 세팅 없이 그래프를 그려보자

import matplotlib.pyplot as plt #그래프 그리는 패키지 import

x = [10,20,30,40,50]
y = [10,20,30,40,50]

plt.plot(x,y)
plt.xlabel('y')
plt.ylabel('X')
plt.show()  #그래프 띄어주기

 

 

예제를 실행 해보면 그려지긴 해도 많이 밋밋하다. 그래서 세팅 몇 개는 해주는 것이 좋다.

 

 

 

 

---

 

산점도 그려보기

이번엔 plot 함수가 아닌 scatter 함수를 사용해서 산점도를 그려보자.

import matplotlib.pyplot as plt #그래프 그리는 패키지 import

x1 = [10,20,30,40,50]
y1 = [10,20,30,40,50]

x2 = [100,200,300,400,500]
y2 = [100,200,300,400,500]

plt.scatter(x1, y1)
plt.scatter(x2, y2, marker = "*")

plt.xlabel('x')
plt.ylabel('y')
plt.show()

 

예제를 실행 시키면 아래와 같은 그래프가 실행된다.

plot 함수와의 차이점이 느껴질 거다.

 

 

 

 

마치며

파이썬은 다른 언어의 비해서 시간 단축을 위한 패키지들을 많이 제공 한다.

matplotlib 패키지도 마찬가지다.

 

C언어로 코딩을 시작한 나에게는 파이썬에서 제공하는 기능들이 놀랍기만 하다.

한번 시간 남는다면 아래 사이트 가서 matplotlib 패키지에 대해 훑어보고 여러가지 그래프를 만들어 보는 것도 좋을 거 같다.

 

https://matplotlib.org/3.5.0/index.html#

Matplotlib documentation — Matplotlib 3.5.0 documentation