[Python] k진수에서 소수 개수 구하기

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

프로그래머스

# k진수에서 소수 개수 구하기
import math
def solution(n, k):
    answer = 0
    
    k_num = ''
    while n > 0:
        n_rest = n % k
        k_num += str(n_rest)
        n = n//k
    k_num = k_num[::-1]
    
    candidate = [word for word in k_num.split('0') if word != '1' and word != '']
    
    for cand in candidate:
        num = int(cand)
        error = 0
        for i in range(2, int(math.sqrt(num))+1):
            if num % i == 0:
                error += 1
                break
        if error == 0:
            answer += 1

    return answer