с задачей
Первый тест вроде работает, а второй нет
Что здесь не так? Почему не решает? Чего возможно не хватает?
Вот сам код:
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))