프로그래머스 - k진수에서 소수 개수 구하기 (L2)

 1def convert10toN(n, k):
 2    result = ''
 3    while n > 1:
 4        result = str(n%k)+ result
 5        n = n // k
 6    return (str(n)+result).lstrip('0')
 7
 8def is_prime(n):
 9    if n <= 1:
10        return False
11    
12    end = int(n**(1/2))+1
13    for i in range(2, end):
14        if n%i == 0:
15            return False
16    return True
17
18
19def solution(n, k):
20    A = convert10toN(n, k)
21    answer = 0
22    for i in A.split('0'):
23        if i == '': continue
24        if is_prime(int(i)):
25            answer += 1
26    
27    return answer