본문 바로가기

컴퓨터과학/Python

[Python 3] matplotlib으로 기초 그래프 그리기

'matplotlib'은 기본으로 내장되어 있지 않기 때문에 사용 전 설치가 선결돼야 한다. cmd에서 pip install matplotlib을 입력하여 설치한다.

 

 

from pylab import plot, show
pm_seoul = [2, 2, 2, 1, 0, 0, 1, 2, 3, 6]
years = range(2007, 2017)
plot(years, pm_seoul)

show()

 

먼저 pylab 모듈에서 plot함수와 show함수를 불러온다. plot함수는 plot(x축 리스트, y축 리스트)과 같이 사용되며, show함수는 그린 그래프를 화면을 통해 보여주는 기능이 있다.

 

3, 4번 줄은 서울특별시 미세먼지 연도별 발령횟수를 2007년부터 2016년까지 pm_seoul로 정의하고 리스트를 만든 모습이다.

 

서울특별시 미세먼지 연도별 발령횟수[각주:1]
출처 : 서울특별시 기후환경본부 - 서울특별시 대기환경정보

 

▲ show함수로 실행한 모습.

 

from pylab import plot, show
pm_seoul = [2, 2, 2, 1, 0, 0, 1, 2, 3, 6]
years = range(2007, 2017)
plot(years, pm_seoul, marker='o')

ozone_seoul = [20,23,14,21,10,6,18,23,4,33]
plot(years, ozone_seoul, marker='+')

from pylab import legend
legend(['PM', 'Ozone'])

from pylab import title, xlabel, ylabel
title('Seoul Air Pollutant Environment Information')
xlabel('Year')
ylabel('Number of alarms')

show()

 

6, 7번 줄은 서울특별시 오존 연도별 발령횟수[각주:2] 그래프도 추가한 모습이다. plot함수엔 marker를 사용하여 각 값에 표시해줄 수 있는데 사용기호로는 ., o, x, +, D, P 등이 있다. D는 다이아몬드, p는 오각형(pentagon)인 식이다. plot(x, y, 'p')처럼 marker를 사용하지 않으면 선은 없고 오각형 점만 찍힌 그래프를 작성할 수 있다.

 

9번 줄에서 불러온 legend는 문자 뜻 그대로 범례를 나타낸다.

12번 줄에서 불러온 title, xlabel, ylabel은 각각 제목, x축 레이블, y축 레이블을 의미한다. 실행하면 다음과 같다.

 

 

아래 디스켓 아이콘을 눌러 그래프를 그림 파일로 저장할 수 있다.

 

 

 

 

 

  1. http://cleanair.seoul.go.kr/alert_year.htm?method=dust [본문으로]
  2. http://cleanair.seoul.go.kr/alert_year.htm?method=ozone [본문으로]