수학과 함께하는 AI기초 Restart D+1 : 생활 데이터 가공하기 - 생각해보기 문제

코드

import csv
import matplotlib.pyplot as plt

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','THU','FRI','SAT','SUN']
hour_title = ['09', '10', '11', '12',
              '13', '14', '15', '16', '17', '18',
              '19', '20', '21'] # 오전 9시부터 밤 9시까지 

avgh_y = [] # 40세 미만 행인의 수의 평균을 위해 선언
avgh_o = [] # 40세 이상 행인의 수의 평균을 위해 선언 

for j in range(0,13):
    day_sum=0
    day_sum_o=0
    # j번째 시간대 총합
    for i in range(0,7):
        day_sum=day_sum+int(a[i][j]['ynum'])
        day_sum_o=day_sum_o+int(a[i][j]['num'])-int(a[i][j]['ynum'])

    avgh_y.append(day_sum/7)
    avgh_o.append(day_sum_o/7)

plt.title('passerby data by age', fontsize=16)
plt.xlabel('hour', fontsize=10)
plt.ylabel('number of passerby by age', fontsize=12)

plt.scatter(hour_title, avgh_y)
plt.plot(hour_title, avgh_y)

plt.scatter(hour_title, avgh_o)
plt.plot(hour_title, avgh_o)
plt.show()