[프로그래머스/Python] Lv 0. 로그인 성공?

⭐ 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/120883

프로그래머스

 

 

⭐ 풀이 코드

def solution(id_pw, db):
    answer = ''
    db_dict = {user[0]:user[1] for user in db}
    
    if id_pw[0] in db_dict:
        if id_pw[1]==db_dict[id_pw[0]]:
            answer = "login"
        else:
            answer = "wrong pw"
    else:
        answer = "fail"
    
    return answer