python: fait `pour i dans obj.func()` réexécute `func` à chaque itération?

Question

disons que je le code suivant:

for a in object.a_really_huge_function(): 
    print a 

Afin d'éviter a_really_huge_function de plusieurs fois en cours d'exécution, je suis habitué à le faire dans d'autres langues:

a_list = object.a_really_huge_function() 
for a in a_list: 
    print a 

Est-ce nécessaire en Python? Est-ce que la partie après in fonctionnera sur chaque boucle?

Answer 1

L'interpréteur python est votre ami.

>>> def some_func(): 
...  print 'in some_func' 
...  return [1, 2, 3, 10] 
... 
>>> for a in some_func(): 
...  print a 
... 
in some_func 
1 
2 
3 
10 

En bref, non, il est appelé une seule fois.

Answer 2

Ce n'est pas nécessaire.

for a in object.a_really_huge_function(): 

appelle a_really_huge_function une seule fois. Le seul avantage de sauvegarder le résultat dans une variable serait si vous appelez la même fonction ailleurs dans votre code.

Si la fonction renvoie une liste, vous pouvez faire mieux en termes d'utilisation de la mémoire en faisant object.a_really_huge_function() retourner un iterator, mais sinon vous allez bien.

Answer 3

Vous pouvez également utiliser des générateurs pour éviter des résultats énormes retour en renonçant à un contrôle après chaque étape de l'algorithme comme suit:

def a_really_huge_fuction(huge_number): 
    i = 0 
    while i < huge_number: 
     yield i # Relinquishes control to caller, resume execution at next line 
     i = i + 1 

Dans ce cas, la fonction est appelée une fois, mais il a sa propagation d'exécution sur huge_number des moments différents. Voir la documentation yield pour plus de détails.