[데이터 분석] 데이터 시각화 - 그래프 그리기, 옵션 추가하기

matplotlib 라이브러리

- 파이썬으로 데이터를 시각화하는 데는 matplotlib 라이브러리를 가장 많이 사용함

- 파이썬에서 2D 형태의 그래프, 이미지 등을 그릴 때 사용하며, 과학 컴퓨팅 연구 분야, 인공지능 연구 분야에서 활용됨

 

라이브러리 불러오기

import matplotlib.pyplot as plt

matplotlib 라이브러리에 속한 pyplot 모듈 불러오기

 

기본 그래프 그리기

import matplotlib.pyplot as plt
plt.plot([10, 20, 30, 40])
plt.show()

- plot()함수 : 직선 또는 꺾은선 형태의 그래프를 그릴 때 사용할 수 있는 명령어

plot() 함수에 입력된 리스트의 값이 y축 값으로 입력되며, x축의 값은 자동으로 0부터 1씩 증가하는 정수로 입력됨

 

---

import matplotlib.pyplot as plt
plt.plot([1,2,3,4],[12,43,25,15])
plt.show()

 

plot() 함수에 두 개의 리스트를 입력했을 때는 첫 번째 리스트가 x축 값, 두 번째 리스트가 y축 값으로 입력됨

 

plot() 함수에 기본 그래프를 그리기 위한 단계

 1. import matplotlib.pyplot as plt : 라이브러리 불러오기

 2. plt.plot([x축 데이터], [y축 데이터]) : plot() 함수에 데이터 입력하기

 3. plt.show() : 그래프 보여주기

 

---

그래프에 옵션 추가하기

 

 ● 그래프에 제목 넣기

제목을 넣는 title() 함수 사용 

import matplotlib.pyplot as plt
plt.title('plotting')
plt.plot([10,20,30,40])
plt.show()

● 그래프에 범례 넣기

- 그래프가 의미하는 바를 구별할 수 있도록 범례 넣기

- 범례는 보통 두 개 이상의 데이터를 표시할 때 사용 ( 2개의 그래프가 하나의 좌표평면에 표현되었을 때 사용 )

- plot() 함수를 사용하여 label이라는 속성의 레이블 값으로 원하는 문자열을 넣어주고, 그래프를 그리기 전에

  legend() 함수를 실행시키면 레이블 값이 범례로 나타남

import matplotlib.pyplot as plt
plt.title('legend')
plt.plot([10,20,30,40], label='asc') #증가를 의미하는 asc 범례
plt.plot([40,30,20,10], label='desc') #감소를 의미하는 desc 범례
plt.legend()
plt.show()

 ● 그래프 색상 바꾸기

color 속성 사용

color 속성 생략 시 자동으로 색상이 설정됨

import matplotlib.pyplot as plt
plt.title('color')
plt.plot([10,20,30,40], color='skyblue', label='skyblue')
plt.plot([40,30,20,10], 'pink', label='pink')
plt.legend()
plt.show()

● 그래프 선 모양 바꾸기

- plot() 함수는 기본적으로 직선으로 그래프를 그림

- 그래프의 선 모양을 바꾸고 싶을 때에는 linestyle 속성에 원하는 선 모양을 지정

- linestyle 속성 대신 ls라고 작성해도 됨

import matplotlib.pyplot as plt
plt.title('linestyle')
plt.plot([10,20,30,40], color='r', linestyle='--', label='dashed')
plt.plot([40,30,20,10], color='g', ls=':', label='dotted')
plt.legend()
plt.show()

 ● 마커 모양 바꾸기

- plot() 함수에 marker 속성을 설정하면 선 형태가 아닌 점 형태로 그래프를 그릴 수 있음

- 이때 색상과 마커 모양을 한번에 설정할 수도 있음

- 마커 모양으로는 선, 점, 동그라미, 삼각형, 별 모양 등이 있음

import matplotlib.pyplot as plt
plt.title('marker')
plt.plot([10,20,30,40], 'r.', label='circle')
plt.plot([40,30,20,10], 'g^', label='triangle up')
plt.legend()
plt.show()

'r.'의 마침표는 점 모양의 마커, 'g^'의 ^는 삼각형 모양의 마커를 의미함

 

 

 

참고서적

모두의 데이터 분석 with 파이썬