Question

Помогите, пожалуйста, с задачей
Первый тест вроде работает, а второй нет



Что здесь не так? Почему не решает? Чего возможно не хватает?


Вот сам код:

def IsPrime(n):

if n % 2 == 0:

return n == 2

d = 3

while d * d <= n and n % d != 0:

d += 2

return d * d > n


b = (int(input()))

a = [i for i in map(int, input().split())]

k = 0

sum = 0

for i in range(len(a)-1):

if IsPrime(a[i]) is True:

sum = sum + a[i]

k=k+1


print(float(sum/k))