[182] 데이터 처리하기 2 연도별 성별 출생아 수 계산하기

def countBirthsBySex():
    ret = []
    for y in range(1880,2016):
        count_f = 0
        count_m = 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]

                tmp = d.split(',')
                sex = tmp[1]
                birth = tmp[2]

                if sex == 'F':
                    count_f += int(birth)
                else:
                    count_m += int(birth)
        ret.append((y,count_f,count_m))
    return ret

result = countBirthsBySex()
with open('birth_by_sex.csv', 'w') as f:
    for y, bf, bm in result:
        data = '%s,%s,%s\n'%(y,bf,bm)
        print(data)
        f.write(data)