[programmers - python] Lv.0: 로그인 성공?

문제 링크

프로그래머스

문제 풀이

def solution(id_pw, db):
    if id_pw in db:
        return "login"
    for data in db:
        if data[0] == id_pw[0]:
            return "wrong pw"
    return "fail"

로그인 성공? 입출력 테스트

*key point: if문과 for문을 적절히 활용하여 주어진 조건에 따라 적합한 return 값을 가지도록 한다.