"인턴이면 당연히 파이썬 쓸 줄 알죠?"

강주원 팀장의 첫 마디였다. 통계학과 3학년 지윤은 고개를 끄덕였다. 파이썬 기초, NumPy, Pandas까지는 수업에서 다뤄봤다.

"그럼 Matplotlib은요?"

"…그게 뭔가요?"

팀장은 웃으며 의자를 당겼다. "오늘 오후까지 설치하고 그래프 하나 그려오면 돼요. 어렵지 않아요."

Matplotlib 설치

Matplotlib은 pip으로 설치합니다. 터미널(또는 Anaconda Prompt)을 열고 다음 명령을 실행합니다.

pip install matplotlib

이미 설치되어 있다면 최신 버전으로 업그레이드할 수 있습니다.

pip install --upgrade matplotlib

설치 후 버전을 확인합니다.

import matplotlibprint(matplotlib.__version__)  # 예: 3.9.2

첫 번째 그래프

설치가 끝났으면 바로 그래프를 그려봅니다. Python 파일이나 Jupyter Notebook을 열고 아래 코드를 실행합니다.

import matplotlib.pyplot as pltx = [1, 2, 3, 4, 5]y = [10, 24, 36, 48, 60]plt.plot(x, y)plt.show()

실행하면 창이 하나 뜨면서 선 그래프가 나타납니다. 지윤의 화면에도 파란 선 하나가 그어졌다.

"이게 다에요?" 지윤이 물었다.

"이게 다예요. Matplotlib은 기본적으로 이게 전부입니다. 여기서 얼마나 정교하게 만드느냐가 스킬이고요."

자주 묻는 질문

Q: 그래프 창이 뜨지 않아요.

스크립트(.py 파일)로 실행할 때는 plt.show()가 없으면 창이 뜨지 않습니다. 반드시 마지막에 plt.show()를 추가하세요.

Q: import matplotlib.pyplot as plt 에서 plt는 뭔가요?

pyplot이라는 서브모듈을 plt라는 짧은 이름으로 불러오는 관례입니다. 전 세계 Matplotlib 코드가 거의 모두 plt를 사용하므로 그대로 따릅니다.

Q: matplotlib 말고 matplotlib.pyplot을 불러오는 이유가 뭔가요?

matplotlib 자체는 라이브러리의 패키지입니다. 실제 그래프를 그리는 함수들은 matplotlib.pyplot에 있습니다. pyplot이 "그림을 그리는 작업대" 역할을 한다고 생각하면 됩니다.