[프로그래머스-Python] 숫자 찾기

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

문제

정수 num과 k가 매개변수로 주어질 때, num을 이루는 숫자 중에 k가 있으면 num의 그 숫자가 있는 자리 수를 return하고 없으면 -1을 return 하도록 solution 함수를 완성해보세요.

제한사항

• 0 < num < 1,000,000
• 0 ≤ k < 10
• num에 k가 여러 개 있으면 가장 처음 나타나는 자리를 return 합니다.

입출력 예

설명

입출력 예 #1

• 29183에서 1은 3번째에 있습니다.

입출력 예 #2

• 232443에서 4는 4번째에 처음 등장합니다.

입출력 예 #3

• 123456에 7은 없으므로 -1을 return 합니다.

def solution(num, k):
    
    num_list = list(map(int, str(num)))
    
    for i, digit in enumerate(num_list):
        if digit == k:
            return i+1            
    return -1

enumerate 함수를 사용하여 num_list를 반복하며 해당하는 인덱스와 함께 숫자를 가져온다.
만약 일치하는 숫자가 없는 경우 -1을 반환한다.