먼저 데이터를 읽어옵니다. 24시간 데이터를 순서대로 저장하기 위해 리스트를 사용합니다. 그리고 승차와 하차 데이터를 각각 저장하기 위한 리스트인 s_in과 s_out을 만듭니다.
import csv f = open('subwaytime.csv') data = csv.reader(f) next(data) next(data) s_in = [0] * 24 # 승차 인원을 저장할 리스트 초기화 s_out = [0] * 24 # 하차 인원을 저장할 리스트 초기화
그리고 모든 역에 대해 시간대별 승차 인원과 하차 인원을 누적해서 더합니다.
for row in data : row[4:] = map(int, row[4:]) for i in range(24) : s_in[i] += row[4+i*2] s_out[i] += row[5+i*2]
마지막으로 두 개의 그래프를 비교하기 위해 꺾은선 그래프를 그려줍니다. 그리고 제목과 범례를 넣어서 한눈에 들어오도록 표현하겠습니다.
이를 전체 코드로 나타내면 다음과 같습니다.