[데이터 시각화] matplotlib, subplot / subplots 여러 개 차트 그리기

1. matplotlib

matplotlib 모듈은 MATLAB과 비슷한 동작을 수행하는 함수의 모음입니다.

 

해당 모듈을 사용하여 간단한 그래프를 그려보겠습니다.

figure() 함수는 그래프를 그려줄 도화지를 만들어 줍니다.

plot() 함수를 이용하여 x,y의 값에 맞게 그래프를 그려줍니다.

import matplotlib.pyplot as plt 

x=[1,2,3]
y=[4,5,6]
plt.figure()
plt.plot(x,y)

이외에도 plot() 함수의 인자로 다양한 것들이 있습니다.

linewidth : 선의 굵기 조절

color : 선 색상 조절

marker : 데이터 위치 표시

linestyle : 선 형태 변경

 

다음으로 그래프 외부 요소에 대해 알아보겠습니다.

title() 함수를 사용하면 그래프의 제목을 설정할 수 있습니다.

xlabel(), ylabel() 함수를 사용하면 x축, y축 명을 설정할 수 있습니다.

xlim(), ylim() 함수를 사용하면 x범위, y범위를 설정할 수 있습니다.

plt.plot(x,y)
plt.title('제목')
plt.xlabel('x축')
plt.ylabel('y축')
plt.xlim([1,2])
plt.ylim([4,5])

여러 데이터를 하나의 차트에 그릴 수 있습니다.

또한 legend() 함수를 사용하면 범례를 사용 할 수 있습니다.

x1=[1,2,3]
y1=[1,2,3]
x2=[1,2,3]
y2=[1,4,7]
plt.plot(x1,y1,color='blue')
plt.plot(x2,y2,color='red')
plt.legend(['data1','data2'])

 

---

2. subplot / subplots 여러 개 차트 그리기

 

subplot과 subplots 함수는 모두 한번에 여러 그래프를 보여주는 용도입니다.

하지만 subplot은 일일이 설정해줘야 하지만, subplots는 한번에 설정 할 수 있습니다.

 

먼저, subplot(행의 수, 열의 수, 해당 그래프가 그려질 위치) 함수를 사용해보겠습니다.

x1=[1,2,3]
y1=[1,2,3]
x2=[1,2,3]
y2=[1,100,200]

plt.subplot(1,2,1)
plt.plot(x1,y1)
plt.title('data1')

plt.subplot(1,2,2)
plt.plot(x2,y2)
plt.title('data2')

 

다음은, subplots()의 사용법입니다.

subplots()에선 두개의 값을 받을 수 있는데 figure 와 axes 값을 받을 수 있다.

fig란 figure로써 전체 subplot을 말합니다.

ax는 axe로써 전체 중 낱낱개를 말합니다.

 

subplots() 함수 안에는 여러 인자가 있습니다.

nrows : 몇행을 만들건지

nrows ; 몇열을 만들것인지

sharex : Xtick의 값을 각각 공유할지 안할지

sharey : Ytick의 값을 각각 공유할지 안할지

fig, axe1 = plt.subplots(nrows=1, ncols=2)

도화지를 그렸으니, 개별의 차트에 그래프를 그려보겠습니다.

axe1[0].plot(x1,y1, color='blue')
axe1[1].plot(x2,y2, color='red')
fig