Ho una funzione del generatore come la seguente:

def myfunct():
  ...
  yield result

Il solito modo di chiamare questa funzione sarebbe:

for r in myfunct():
  dostuff(r)

La mia domanda, c'è un modo per ottenere solo un elemento dal generatore ogni volta che mi piace? Ad esempio, vorrei fare qualcosa del tipo:

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

Crea un generatore usando

g = myfunct()

Ogni volta che desideri un oggetto, usa

next(g)

(o g.next()in Python 2.5 o precedenti).

Se il generatore esce, si alza StopIteration. Puoi prendere questa eccezione, se necessario, o usare l' defaultargomento per next():

next(g, default_value)

Per selezionare solo un elemento di un generatore utilizzare breakin forun'istruzione, oppurelist(itertools.islice(gen, 1))

Secondo il tuo esempio (letteralmente) puoi fare qualcosa del tipo:

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

Se vuoi " ottenere un solo elemento dal generatore [una volta generato] ogni volta che mi piace " (suppongo che il 50% sia l'intenzione originale e l'intenzione più comune) quindi:

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

In questo modo è generator.next()possibile evitare l' uso esplicito di e la gestione di fine input non richiede la gestione (criptica) delle StopIterationeccezioni o confronti di valori di default aggiuntivi.

La sezione else:of forstatement è necessaria solo se vuoi fare qualcosa di speciale in caso di end-of-generator.

Nota su next()/ .next():

In Python3 il .next()metodo è stato rinominato .__next__()per una buona ragione: è considerato di basso livello (PEP 3114). Prima di Python 2.6 la funzione incorporata next()non esisteva. Ed è stato anche discusso di passare next()al operatormodulo (che sarebbe stato saggio), a causa della sua rara necessità e discutibile inflazione dei nomi predefiniti.

L'uso next()senza impostazione predefinita è ancora una pratica di livello molto basso: lanciare il criptico StopIterationcome un fulmine a ciel sereno nel normale codice dell'applicazione. E l'utilizzo next()con sentinel predefinito - che dovrebbe essere l'unica opzione per un accesso next()diretto builtins- è limitato e spesso dà ragione a strane logiche / leggibilità non pitoniche.

In conclusione: usare next () dovrebbe essere molto raro, come usare le funzioni del operatormodulo. Usando for x in iterator, islice, list(iterator)e altre funzioni di accettare un iteratore senza soluzione di continuità è il modo naturale di usare iteratori sul livello di applicazione - e quasi sempre possibile. next()è di basso livello, un concetto in più, invisibile - come mostra la domanda di questo thread. Mentre ad esempio l'uso di breakin forè convenzionale.

Non credo che ci sia un modo conveniente per recuperare un valore arbitrario da un generatore. Il generatore fornirà un metodo next () per attraversare se stesso, ma l'intera sequenza non viene prodotta immediatamente per risparmiare memoria. Questa è la differenza funzionale tra un generatore e un elenco.

Per quelli di voi che scansionano queste risposte per un esempio di lavoro completo per Python3 ... beh ecco qui:

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)

Questo produce:

1001
1002
1003

Il generatore è una funzione che produce un iteratore. Pertanto, una volta che si dispone dell'istanza iteratore, utilizzare next () per recuperare l'elemento successivo dall'iteratore. Ad esempio, utilizzare la funzione next () per recuperare il primo elemento e successivamente utilizzare for inper elaborare gli elementi rimanenti:

# 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()

assicurati di rilevare l'eccezione generata dopo che è stato preso l'ultimo elemento

Credo che l'unico modo sia ottenere un elenco dall'iteratore e quindi ottenere l'elemento desiderato da tale elenco.

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