[Python] CSV 파일 사용하기

csv 라이브러리 설치

  파이썬 코드에서 그래프를 그리기 위해서는 csv이라는 라이브러리를 설치해야 한다. 아래와 같은 명령을 콘솔 창에 입력하면 설치가 가능하다. 

pip install csv

만약에 pip 명령어를 찾을 수 없다면 install 시에 환경 변수를 등록하지 않았기 때문이다.

python3.8의 경우는 아래와 같다.

C:\Users\계정\AppData\Local\Programs\Python\Python38
C:\Users\계정\AppData\Local\Programs\Python\Python38\Scripts

환경 변수에 아래와 같은 경로를 추가하면 된다.

 

Python을 사용한 CSV 파일 읽어오기

import csv
a = [[],[],[],[],[],[],[]]

with open('passby_data.csv', 'r') as f :
    reader = csv.DictReader(f)
    i = j = 0
    for row in reader :
        a[i].append(row)
        j = j + 1
        if (j%24 == 0) :
            i = i + 1

day_title = ['MON', 'TUE', 'WED', 'THR', 'FRI', 'SAT', 'SUN']
hour_title = ['01', '02', '03', '04', '05', '06', \
              '07', '08','09', '10', '11', '12', \
              '13', '14','15', '16', '17', '18', \
              '19', '20','21', '22', '23', '24',]

#시간대별로 주간 평균 구하기
avgh = []
for j in range(0, 24) :
    day_sum = 0
    #j번째 시간대 주간 총합
    for i in range(0, 7) :
        day_sum = day_sum + int(a[i][j]['num'])

    avgh.append(day_sum/7)

#시간대별 평균 유동 인구 출력하기
for j in range(0, 24) :
    print("[~{0}:00]: {1:4}".format(hour_title[j], int(avgh[j])))

passby_data.CSV

0.00MB

Python IDLE Shell로 실행하면 아래와 같은 결과를 얻을 수 있다.