숫자 찾기

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

내 코드

def solution(num, k):
    answer = 0
    str_num = str(num)
    str_k = str(k)
    if str_k in str_num:
        answer = str_num.index(str_k)+1
    else:
        answer = -1
    return answer

예시 코드

def solution(num, k):
    return -1 if str(k) not in str(num) else str(num).find(str(k)) + 1