[프로그래머스] 실패율(easy, kakao)

마지막 result부분에 session이 0일때 케이스를 고려해줘야한다.

• 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다. (이 조건문을 고려해줘야함)

def solution(N, stages):
    answer = []
    temp = []
    for i in range(1,N+1):
        cnt = 0
        session = 0
        result = 0
        for j in stages:
            if j >= i:
                session += 1
            if i-1<j<= i:
                cnt += 1
                
        if session == 0:
            result = 0
        else:
            result = cnt/session
        temp.append((i, result))

    temp.sort(key=lambda t:t[1], reverse=True)
    print(temp)
    for i in temp:
        answer.append(i[0])
    return answer