[Python] 소수 만들기

https://school.programmers.co.kr/learn/courses/30/lessons/12977?language=python3 

프로그래머스

# 소수 만들기

from itertools import combinations

def solution(nums):
    answer = 0
    
    sum_list = []
    for i in combinations(nums, 3):
        sum_list.append(sum(i))
        
    for j in sum_list:
        error = 0
        if j == 1:
            continue
        else:
            for k in range(2,j):
                if j % k == 0:
                    error += 1
                    break
                    
        if error == 0:
            answer += 1
            
    return answer