[python]L1: Bartender, drinks!

https://www.codewars.com/kata/568dc014440f03b13900001d/train/python

Codewars: Train your coding skills

입력	출력
Jabroni	Patron Tequila
School Counselor	Anything with Alcohol
Programmer	Hipster Craft Beer
Bike Gang Member	Moonshine
Politician	Your tax dollars
Rapper	Cristal
기타	Beer

이 문제는 특정한 값을 입력받았을 때 정해진 값을 출력하는 것입니다.

 

if, elif와 else를 사용해도 되지만 dictionary를 이용해서 문제를 풀었습니다.

 

여기서 중요한 점은 입력값에 대소문자가 섞여 있어도 출력을 해야 하는 것입니다.

이것을 해결하기 위해서는 입력값을 소문자나 대문자로 모두 변경해도 앞글자만 대문자로 바꾸어도 됩니다.

 

제 코드를 보시려면 더보기를 누르시면 됩니다.

L1: Bartender, drinks!

# My Code
def get_drink_by_profession(param):
    dic = {'Jabroni': 'Patron Tequila',
         'School Counselor': 'Anything with Alcohol',
         'Programmer': 'Hipster Craft Beer',
         'Bike Gang Member': 'Moonshine',
         'Politician': 'Your tax dollars',
         'Rapper': 'Cristal'}
    return dic[param.title()] if param.title() in dic else 'Beer'
 
if __name__=='__main__':
    print(get_drink_by_profession("jabrOni"))
    print(get_drink_by_profession("scHOOl counselor"))
    print(get_drink_by_profession("prOgramMer"))
    print(get_drink_by_profession("bike ganG member"))
    print(get_drink_by_profession("poLiTiCian"))
    print(get_drink_by_profession("rapper"))
    print(get_drink_by_profession("pundit"))
    print(get_drink_by_profession("Pug"))