그럼 처음부터 과정 5까지 프로그램으로 옮겨 봅시다. 필요한 라이브러리를 추가하고 알고리즘 옆의 코드를 차례대로 적습니다. 리스트 weekday와 weekend에 값이 제대로 들어갔는지 확인하기 위해 마지막에 print()를 추가합니다.
import datetime
weekday = [0 for i in range(24)]
weekend = [0 for i in range(24)]
for row in data:
if row[2] == dong_code:
time, p = row[1], row[3]
year, mon, day = int(row[0][:4]), int(row[0][4:6]), int(row[0][6:])
num = datetime.date(year, mon, day).weekday()
if num < 5:
weekday[time] += p
else:
weekend[time] += p
# 2019년 12월의 주중/주말 일수 구하기
weekday_cnt, weekend_cnt = 0, 0
for i in range(1, 32):
if datetime.date(2019, 12, i).weekday() < 5:
weekday_cnt += 1
else:
weekend_cnt += 1
print('2019년 12월의 주중 일수 =', weekday_cnt, '주말 일수 =',weekend_cnt)
weekday = [w/weekday_cnt for w in weekday]
weekend = [w/weekend_cnt for w in weekend]
print('주중 인구:', weekday)
print('주말 인구:', weekend)
실행결과
2019년 12월의 주중 일수 = 22 주말 일수 = 9
주중 인구: [36348.0879, 35284.529854545464… 45000.37916363635, 40292.146422727266]
주말 인구: [38933.31007777777, 37376.29128888888… 40075.01161111111, 37075.203344444446]