[Python] 데이터 시각화

이전 글에서 csv 파일 데이터를 다뤄보았다.

https://yn98.tistory.com/130

[Python] csv 파일 데이터 다뤄보기

 

오늘은 위 데이터를 시각화해보려고 한다.

시각화하기 위해서는, python에서는 matplolib 라이브러리의 pyplot 모듈을 가져와야 한다.

 

matplolib는 외부 라이브러리이기 때문에, 다운로드를 해야한다.

cmd 창에 아래와 같은 명령어를 입력해서 다운로드를 하면 된다.

 

pip install matplotlib

 

다운로드 후, 상단에 import문을 작성한다.

import csv
import matplotlib.pyplot as plt

 

이제, 데이터를 시각화할 수 있는 기반을 다 다졌다고 보면 된다.

import csv

import matplotlib.pyplot as plt

file_path = "test.csv"

dates = []
temps = []
with open(file_path,mode='r') as file :
    reader = csv.reader(file) # 자바에서는 패키지, 파이썬에서는 모듈

    header = next(reader)

    for row in reader :# 향상된 for문은    for v in 집합
        a = row[2] ## 12월인 데이터만 출력하고싶어~~
        b = row[-2] # 최저기온
         
        if a.startswith("Dec") :#if a가 Dec 시작하면 :
            print(a, b) # row가 list 타입이라는 사실도 확인 가능
            dates.append(a)
            temps.append(float(b))

plt.plot(dates,temps,marker='o',color='r',linestyle=':') # 선
##plt.plot(dates,temps) #그래프를 그리고(x축, y축) # 기본

plt.title("Temperature in December", fontsize=15)
plt.xlabel("Date", fontsize = 10)
plt.ylabel("Temperature", fontsize=10)

plt.show() #그린 그래프를 화면에 출력해줘()

기존 코드에서, plt 모듈을 사용하면, 위와 같이 시각화를 할 수 있다.