프로그래머스 소수 찾기 (level 2)

programmers.co.kr/learn/courses/30/lessons/42839#qna

코딩테스트 연습 - 소수 찾기

import itertools
def solution(numbers):
    num = []
    for i in range(len(numbers)):
        num.extend(list(set(list(map(''.join, itertools.permutations(numbers,i+1))))))

    num=list(set([int(i) for i in num]))
    answer = len(num)
    for i in range(len(num)):
        if num[i] == 0 or num[i] == 1:
            answer -= 1
        if num[i] > 2:
            for j in range(2, int(num[i]**0.5)+1):
                if num[i]%j == 0:
                    answer -= 1
                    break  
    return answer