У меня есть генератор в моей функции Python; как я могу вернуть измененный список?

Question

Немного фона: Я пытался закодировать алгоритм «Решетка Эратосфена». По просьбе некоторых тонких (и очень терпеливых) программистов в чат-комнате StackOverflow Python я прочитал некоторую информацию о функции enumerate() и нашел способ включить ее в свой код (да, я здесь очень новичок). До сих пор мой (работоспособные, возвращаю ожидаемый отклик) код выглядит следующим образом:

def SieveErat(n): 
    numbers = [False]*2+[True]*(n-1) 

    for index, prime_candidate in enumerate(numbers): 
     if prime_candidate == True: 
      yield index 
      for x in xrange(index*index, n, index): 
       numbers[x] = False 

primes = [] 

for x in SieveErat(150000): 
    primes.append(x) 

print primes[10002] 

Излишне говорить, что функция enumerate() делает кодирование это гораздо, гораздо менее громоздкими, чем бы то ни были вложенные циклы у меня было раньше. Тем не менее, я боюсь, что я не понимая, что-то о enumerate(), потому что, когда я попытался сократить этот код, включив в эту директиву функции, я продолжал получать errors- именно

File "SievErat.py", line 13 
     return numbers 
SyntaxError: 'return' with argument inside generator 

Я также попытался добавления все True элементов в списке numbers в инициализированный список primes, но не нашел удачи.

Любые советы и советы были бы очень приветствуемыми.

Answer 1

Это не имеет ничего общего с enumerate вы пытаетесь вернуть что-то в генераторе, который, прежде чем питона 3,3 является незаконным, и from 3.3+ it means something entirely different.

Я рекомендую вам оставить функцию генератора, если вы можете использовать его без необходимости список назад, и если вы хотите получить результаты списка, то просто позвоните list() на возвращаемом значении:

primes = list(SieveErat(150000)) #this replaces loop with .append 

Но чтобы понять, что пошло не так, если ваша функция все еще имеет yield заявления, в нем, то он должен вернуть объект генератора , если вы не хотите, чтобы он возвращался гп объект генератора затем снимите yield заявления все вместе:

def SieveErat(n): 
    numbers = [False]*2+[True]*(n-1) 

    for index, prime_candidate in enumerate(numbers): 
     if prime_candidate == True: 
      #yield index #no yield statement if you want to return the numbers list 
      for x in xrange(index*index, n, index): 
       numbers[x] = False 

    return numbers #return at end 

Однако это будет потом возвращать список True и False вместо чисел, которые, правда, вы можете вместо этого провести отдельный список всех простых чисел и .append к нему каждый раз, когда вы что-то yield:

def SieveErat(n): 
    numbers = [False]*2+[True]*(n-1) 
    result = [] #start with no results 
    for index, prime_candidate in enumerate(numbers): 
     if prime_candidate == True: 
      results.append(index) #instead of yield index 
      for x in xrange(index*index, n, index): 
       numbers[x] = False 
    return results 

Но это чувствует, как шаг назад от генератора, лично я бы просто сохранить то, что вы отправили и бросили результат в list.