matplotlib(plt)에서두개의 선 그래프를 겹쳐 그리기 & plt.legend( )

subplot은 각각 칸을 나누어서 해당하는 부분에 내가 원하는 그래프를 넣을 수 있는 방법이다.

하지만, 데이터 분석에서 시각화를 하는 입장에서는 두 그래프를 한곳에 두고 비교를 하고 싶은 경우 겨쳐 그리고 싶은 경우가 종종 있다. 이런 경우에는 생각보다 간단하게 코드로 구현할 수 있다.

 

import matplotlib.pyplot as plt

plt.plot(x_1, y_1, x_2, y_2)
plt.show()

위와 같이 그래프 1에 대한 x와 y에 대한 정보를 입력하고, 바로 뒤에 별도의 parameter를 지정할 필요 없이, 순서대로 입력을 해주면 된다. 2개의 직선 뿐만 아니라 더 많은 그래프를 한 곳에 겹쳐 그리는 경우에도 위와 같은 방식으로 뒤에 계속해서 이어 붙이면 겹쳐 그려지는 것을 확인할 수 있다.

 

그래프에서 범례(legend)를 사용하는 기본적인 방법

 

그래프를 겹쳐 그리게 되고, 더 복잡한 직선들을 나타내야 하는 경우에는 범례(legend)의 작성이 필요하다. 가장 기본적인 범례를 작성하는 방법은 plt.legend( )를 사용해서 그 안에 리스트를 입력하고 순차적으로 직선의 이름을 정해주면 된다.

 

예시 > 3개의 직선을 겹쳐 그리기 & 범례(legend) 설정하기 

 

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
x = pd.Series([1,2,3,4,5])
y_1 = 5 * x + 1
y_2 = 2 * x + 4
y_3 = 3 * x + 3

plt.plot(x, y_1, x, y_2, x, y_3)
plt.legend(['line1','line2','line3'])
plt.show()

이처럼 그래프를 겹쳐 그리는 것이 가능하다.