여기서 9, 10행은 숫자 데이터를 한꺼번에 처리하는 부분으로 for 반복문을 사용하여 row의 3번 인덱스 값부터 끝 번 인덱스 값까지 저장합니다.
for i in row[3:] : home.append(int(i))
이 코드를 Unit 13에서 배운 숫자를 다루는 유용한 라이브러리인 numpy를 사용하여 수정하겠습니다.
import numpy as np import csv f = open('age.csv') data = csv.reader(f) next(data) name = input('인구 구조가 알고 싶은 지역의 이름(읍면동 단위)을 입력해주세요 : ') for row in data : if name in row[0] : home = np.array(row[3:], dtype=int) print(home)