[Python]사용자 입력을 받아 Dictionary 검색

사용자 입력을 받아 Dictionary에서 데이터를 검색하는 방법을 알아보자.

d = {'USA': 36, 'Germany': 17,'France':32, 'Australia': 77, 'South Africa': 99, 'India': 108, 'South Korea': 200}

# 방법1
def search_country_code(code):   
    if d.get(code) != None:
        return d.get(code)
    if d.get(code.title()) != None:
        return d.get(code.title())
    if d.get(code.upper()) != None:
        return d.get(code.upper())
    return "찾는 국가코드가 없습니다."

code = input("국가 코드를 입력하세요 : ")
print(search_country_code(code))

# 방법2
def search_country_code2(code):   
    try:
        new_d = dict((new_k.lower(), new_v) for new_k, new_v in d.items())        
        return new_d[code]        
    except:
        return "찾는 국가코드가 없습니다."
    
code = input("국가 코드를 입력하세요 : ").lower()
print(search_country_code(code))    

# 방법3
def search_country_code3(code):   
    new_d = dict((new_k.lower(), new_v) for new_k, new_v in d.items())        
    return new_d.get(code, "찾는 국가코드가 없습니다.")
    
code = input("국가 코드를 입력하세요 : ").lower()
print(search_country_code(code))    

## 결과
국가 코드를 입력하세요 : usa
36

You may also like...

답글 남기기

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다