range() 함수를 사용하면 row 리스트의 인덱스를 좀더 쉽게 다룰 수 있습니다. 남성과 여성의 합계를 각각 구한 후에는 size 리스트에 성별 인구 합계를 추가하여 결과를 출력하도록 코드를 수정하였습니다.
import csv f = open('gender.csv') data = csv.reader(f) size = [] name = input('찾고 싶은 지역의 이름을 알려주세요 : ') for row in data : if name in row[0] : m = 0 f = 0 for i in range(101) : m += int(row[i+3]) # 3 ~ 103번까지 데이터 더하기 f += int(row[i+106]) # 106 ~ 206번까지 데이터 더하기 break size.append(m) # 남성 합계 데이터 리스트에 추가 size.append(f) # 여성 합계 데이터 리스트에 추가 print(size)
실행 결과
찾고 싶은 지역의 이름을 알려주세요 : 제주특별자치도──▶ 입력 [335813, 331524]
TIP
여러분의 실행 결과는 책과 다를 수 있습니다.