행정동명과 행정동코드 연결하기

그럼 두 번째 함수를 만들어 봅시다. 행정동명을 입력받으면 행정동코드를 알려 주는 dong_search() 함수입니다. dong_search() 함수는 행정동명을 입력받아 행정동코드를 반환합니다. 매개변수 dong_name으로 행정동명을 입력받고 함수 내부에서 찾은 행정동코드는 code에 저장하겠습니다(변경 전 코드는 '11.2.1 데이터  파일일 읽고 행정동명과 행정동코드 연결하기'에 있습니다).

def dong_search(dong_name):
    for row in code_data:
        if row[-1] == dong_name:
            code = row[1]
    print(dong_name, '-', code, '을(를) 분석합니>!')
    return code

프로그램에서 dong_search() 함수를 호출할 때는 인자로 행정동명을 넣어 호출하고, 반환값으로 받은 행정동코드를 저장하면 됩니다.

data, code_data = file_open()

input_name = input('핫플레이스가 위치한 행정동을 입력하세요. --> ')
input_code = dong_search(input_name)

사용자에게 행정동명을 입력받아 input_name에 저장하고, dong_search() 함수를 호출할 때 input_name을 인자로 넣습니다. 함수 실행결과로 반환되는 값은 input_code에 저장합니다. 앞으로 진행할 인구 데이터 분석은 모두 input_name과 input_code에 저장된 지역을 기준으로 합니다.