[matplotlib] 그래프의 y축 범위 지정하는 방법

안녕하세요. 비스카이비전입니다.

 

오늘은 matplotlib로 그린 그래프의 y축 범위를 지정하는 방법에 대해 살펴보도록 하겠습니다. matplotlib는 데이터에 맞게 적절한 범위로 그래프를 그려줍니다. 

 

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame([[1, 5], [2, 2], [3, 7], [4, 9], [5, 6]])

plt.plot(df[0], df[1])
plt.grid(True)
plt.title('title')
plt.xlabel('xlabel')
plt.ylabel('ylabel')
plt.show()

 

 

matplotlib는 주어진 데이터에 근거하여 y값의 범위를 약 1.x부터 9.x로 그려준 것을 확인할 수 있습니다. 

 

matplotlib 그래프 y축 범위 지정하는 법

그러나 어떤 경우에는 그 범위를 사용자가 정하고 싶을 때가 있습니다. 저는 이 그래프의 y축 범위가 -5부터 20까지 나오도록 해보겠습니다. 위 코드에 plt.ylim(-5, 20)을 추가해주시면 됩니다.

 

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame([[1, 5], [2, 2], [3, 7], [4, 9], [5, 6]])

plt.plot(df[0], df[1])
plt.grid(True)
plt.title('title')
plt.xlabel('xlabel')
plt.ylabel('ylabel')
plt.ylim(-5, 20)
plt.show()

 

 

이렇게 그리고 보니까 값의 변동이 비교적 작게 느껴지죠? 만약 y 값이 보통은 -5에서 20 사이에 나오는 상황이라는 것을 알고 있는 상황이라면, 이렇게 지정해서 그리시는 것이 조금 더 정보 전달에 도움이 됩니다. 그렇지 않고 위 그래프와 같이 디폴트로 그리시면 y 값의 변동이 상당히 크다는 인상을 줄 수 있습니다.