Mam funkcję generatora, jak poniżej:

def myfunct():
  ...
  yield result

Zwykłym sposobem wywołania tej funkcji byłoby:

for r in myfunct():
  dostuff(r)

Moje pytanie: czy istnieje sposób na uzyskanie tylko jednego elementu z generatora, kiedy tylko chcę? Na przykład chciałbym zrobić coś takiego:

while True:
  ...
  if something:
      my_element = pick_just_one_element(myfunct())
      dostuff(my_element)
  ...

Utwórz generator za pomocą

g = myfunct()

Za każdym razem, gdy chcesz przedmiot, użyj

next(g)

(lub g.next()w Python 2.5 lub niższym).

Jeśli generator wyjdzie, podniesie się StopIteration. W razie potrzeby możesz złapać ten wyjątek lub użyć defaultargumentu, aby next():

next(g, default_value)

Do wybrania tylko jednego elementu generatora użyj breakw forinstrukcji lublist(itertools.islice(gen, 1))

Zgodnie z twoim przykładem (dosłownie) możesz zrobić coś takiego:

while True:
  ...
  if something:
      for my_element in myfunct():
          dostuff(my_element)
          break
      else:
          do_generator_empty()

Jeśli chcesz „ dostać tylko jeden element z generatora [raz wygenerowanego] , kiedy tylko chcę ” (przypuszczam, że 50% to pierwotna intencja i najczęstsza intencja):

gen = myfunct()
while True:
  ...
  if something:
      for my_element in gen:
          dostuff(my_element)
          break
      else:
          do_generator_empty()

W ten sposób generator.next()można uniknąć jawnego użycia , a obsługa końca wejścia nie wymaga (tajemniczej) StopIterationobsługi wyjątków lub dodatkowych porównań wartości domyślnych.

else:Z forsekcji oświadczenie jest potrzebne tylko wtedy, gdy chcesz zrobić coś specjalnego w przypadku wycofanych z generatora.

Uwaga na next()/ .next():

W Python3 .next()zmieniono nazwę tej metody .__next__()na dobry powód: uważany za niski poziom (PEP 3114). Przed Pythonem 2.6 wbudowana funkcja next()nie istniała. Dyskutowano nawet o przejściu next()do operatormodułu (co byłoby mądre) ze względu na jego rzadką potrzebę i wątpliwą inflację wbudowanych nazw.

Używanie next()bez domyślnych ustawień jest nadal bardzo niskim poziomem praktyki - rzucanie zagadkami StopIterationjak grom z jasnego nieba w normalnym kodzie aplikacji otwarcie. Korzystanie next()z domyślnego wartownika - który najlepiej powinien być jedyną opcją dla next()bezpośredniego builtinswejścia - jest ograniczone i często daje powód dziwnej nie pytonicznej logiki / zdolności odczytu.

Konkluzja: Używanie next () powinno być bardzo rzadkie - podobnie jak korzystanie z funkcji operatormodułu. Korzystanie for x in iterator, islice, list(iterator)oraz inne funkcje przyjmowania iterator bezproblemowo jest naturalnym sposobem korzystania iteratory na poziomie aplikacji - i dość zawsze możliwe. next()jest niski poziom, dodatkowa koncepcja, nieoczywista - jak pokazuje pytanie tego wątku. Podczas gdy np. Używanie breakw forjest konwencjonalne.

Nie sądzę, aby istniał wygodny sposób na uzyskanie dowolnej wartości z generatora. Generator zapewni następną () metodę przejścia przez siebie, ale pełna sekwencja nie jest tworzona natychmiast, aby zaoszczędzić pamięć. Na tym polega funkcjonalna różnica między generatorem a listą.

Dla tych, którzy przeglądają te odpowiedzi w celu uzyskania kompletnego przykładu dla Python3 ... cóż, proszę:

def numgen():
    x = 1000
    while True:
        x += 1
        yield x

nums = numgen() # because it must be the _same_ generator

for n in range(3):
    numnext = next(nums)
    print(numnext)

To daje:

1001
1002
1003

Generator to funkcja tworząca iterator. Dlatego po utworzeniu instancji iteratora użyj next (), aby pobrać następny element z iteratora. Na przykład użyj funkcji next (), aby pobrać pierwszy element, a następnie użyj for indo przetworzenia pozostałych elementów:

# create new instance of iterator by calling a generator function
items = generator_function()

# fetch and print first item
first = next(items)
print('first item:', first)

# process remaining items:
for item in items:
    print('next item:', item)

generator = myfunct()
while True:
   my_element = generator.next()

upewnij się, że złapałeś wyjątek zgłoszony po zajęciu ostatniego elementu

Uważam, że jedynym sposobem jest uzyskanie listy z iteratora, a następnie pobranie żądanego elementu z tej listy.

l = list(myfunct())
l[4]