Python Primes Array Confusion

Question

У меня есть небольшой вопрос об этой простой функции проверки python. Это действительно глупо, но что делает [2] в for n in [2]?

Я понимаю формулу для проверки на простые числа, без проблем, но не могу понять, что такое [2] для.

от: http://en.wikibooks.org/wiki/Python_and_Math

from math import ceil 
def prime(input): 
    for n in [2] + range(3, int(ceil(input**0.5)), 2): 
     if input%n == 0: 
      return False 
    return True 

Спасибо за ответ на мой немой вопрос.

Answer 1

Это просто список того:

>>> range(3, 10, 2) 
[3, 5, 7, 9] 
>>> [2] + range(3, 10, 2) 
[2, 3, 5, 7, 9] 

то есть они хотят, чтобы проверить делимость на 2, но, очевидно, не нужно проверять делимость на 4, 6, 8 ... так что с этой операцией «добавление» a 2 в начало списка нечетных чисел.

К сожалению, эта функция не работает - она ​​возвращает prime(2) == False и prime(1) == True, которые оба неправильно.