길이에 따른 연산

정수가 담긴 리스트 num_list가 주어질 때,
리스트의 길이가 11 이상이면 리스트에 있는 모든 원소의 합을
10 이하이면 모든 원소의 곱을
return하도록 solution 함수를 완성해주세요.

내 코드

def solution(num_list):
    answer = 0
    if len(num_list) >= 11:        
        for i in num_list:
            answer = answer + i
    else:
        answer = 1
        for i in num_list:
            answer = answer * i           
    return answer

예시 코드

from math import prod
def solution(num_list):
    return sum(num_list) if len(num_list)>=11 else prod(num_list)