Sind lokale Variablen in Generatorfunktionen nicht verfügbar?

Question

ich Python lernen und lief dieses Stück Code in der Python-Konsole früher heute:

num = 0 

def generator(): 
    while True: 
     yield num 
     num += 1 

for i in generator(): 
    if i > 5: break 
    print(i) 

Es hat einen

UnboundLocalError: local variable 'num' referenced before assignment warf ich wieder schrieb den Code und diese Version gearbeitet:

def generator(): 
    num = 0 
    while True: 
     yield num 
     num += 1 

for i in generator(): 
    if i > 5: break 
    print(i) 

Meine Frage ist das: Können Sie nicht lokale Variablen in Generatorfunktionen verwenden, wie Sie mit regulären Funktionen können?

Answer 1

Im ersten Code versuchen Sie, num als globale Variable zu verwenden: Es ist außerhalb der Funktion definiert.

Wenn das Programm num += 1 erreicht, versuchen Sie dann, eine lokale Variable zuzuweisen, die denselben Namen wie die globale hat, die Sie zuvor referenziert haben. Da einer lokalen Variablen mit dem Namen num noch kein Wert zugewiesen wurde, schlägt das Inkrement fehl.

Wenn Sie eine globale Variable innerhalb einer Funktion zuweisen oder inkrementieren wollen, muss sie mit global num deklariert werden (normalerweise gleich zu Beginn der Funktion). Das hat nichts mit Generatoren zu tun, es würde auch für normale Funktionen versagen.

Answer 2

Wie UnholySheep has commented ist die num Variable nicht lokal innerhalb der generator() Funktion definiert.

Gerade Marke, die Sie aus dem globalen Namensraum in der generator() Funktion zu verwenden, mit dem global Stichwort:

num = 0 

def generator(): 
    global num 
    while True: 
     yield num 
     num += 1 

for i in generator(): 
    if i > 5: break 
    print(i) 

... und alles sollte in Ordnung sein.

Hinweis: Das Problem hängt nicht mit der Art der Funktion (Generator einschließlich yield) zusammen, die Sie verwenden.