[181] 데이터 처리하기 1 연도별 출생아 수 계산하기

def countBirths():
    ret = []
    for y in range(1880, 2016):
        count = 0
        filename = 'names/yob%d.txt'%y
        with open(filename,'r') as f:
            data = f.readlines()
            for d in data:
                if d[-1] == '\n':
                    d = d[:-1]

                birth = d.split(',')[2]
                count += int(birth)
        ret.append((y,count))
    return ret

result = countBirths()
with open('birth_by_year.csv','w') as f:
    for year, birth in result:
        data = '%s,%s\n'%(year,birth)
        print(data)
        f.write(data)