[프로그래머스] 연습문제 > 소수만들기

from itertools import combinations
import math

def is_prime_num(n):
    for i in range(2, int(math.sqrt(n))+1):
        if n % i == 0:
            return False
    return True

def solution(nums):
    answer = []
    select_nums = list(combinations(nums, r=3))
    
    for i, j, k in select_nums:
        sum = i+j+k
        if is_prime_num(sum):
            answer.append(sum)
    
    return len(answer)