데이터 시각화 기초 - 그래프 그리기

데이터를 시각화하는데 가장 많이 사용하는 matplotlib 라이브러리를 활용하여 그래프를 그리는 방법

*pyplot는 공학용 SW MATLAB와 사용법이 유사하다.

0. Matplotlib 설치

 우선 Matplotlib을 설치

pip install matplotlib

---

1. 기본 그래프 그리기

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

plt.plot()에서 X축 데이터 값만 넣으면 기본적으로 Y축 값이 자동으로 생성된다.

X축 데이터 값만 제공된 경우 Y축 데이터 값으로 인덱스를 사용하며 이는 Y축 값이 0부터 시작하는 정수 시퀀스로 설정이 된다는 것이다.

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

plt.plot([X축 데이터], [Y축 데이터]) 로 각 축에 해당하는 값을 입력하여 나타낸다.

---

2. 그래프에 옵션 추가하기

 2-1 그래프에 제목 넣기

import matplotlib.pyplot as plt
plt.title('제목을 입력하세요')
plt.plot([10,20,30,40])
plt.show()

 plot.title('Title') 을 작성하면 위 와 같이 그래프에 제목이 생성된다.

---

 2-2 범례 넣기

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

 두 개 이상의 데이터를 표시할 때 사용하는 범례는 plot()함수에 label이라는 속성의 레이블 값으로 원하는 문자열을 넣고 plt.legend()함수를 실행 하면 레이블 값이 범례로 나타남.

---

 2-3. 그래프 색상 바꾸기

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()

 그래프의 선 색상을 원하는 색으로 바꾸려면 color 속성을 추가하여 바꾸면 된다.

 *기본적인 색은 약자로 표시 할 수 있다. (r=red, g=green, b=blue, k=black, y=yellow)

---

  2-4 그래프 선 모양 바꾸기

import matplotlib.pyplot as plt
plt.title('linestyle')
#빨간색 dashed 그래프
plt.plot([10,20,30,40], color='r', linestyle='--', label='dashed')
#초록색 dotted 그래프
plt.plot([40,30,20,10], color='g', ls=':', label='dotted')
plt.legend()
plt.show()

plo()함수는 기본적으로 직선으로 나타내지만 linestyle 속성을 이용하여 원하는 선 모양으로 바꿀 수 있다.

---

 2-5. 마커 모양 바꾸기

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()

  plot() 함수에서 maker 속성을 설정하면 선이 아닌 점 형태의 그래프를 그린다. 

 

 

참고 서적 : 모두의 데이터분석 with파이썬
Matplotlib란 : https://sh9406.tistory.com/7