Compréhension de liste avec calcul coûteux

Question

J'ai donc quelques compréhensions de liste qui ressemblent à ceci.

li = [some_computation(x) for x in y if some_computation(x)] 

Où je veux que les valeurs qui sont truthy par rapport à some_computation.

1. Est-ce que cela calcule deux fois some_computation(x) comme la compréhension de la liste semble le faire? Si some_computation est cher, je vraiment ne veulent pas veulent cela. (Je pense que c'est probablement le cas :()
2. Y at-il une meilleure façon de le faire? Peut-être de le casser dans une boucle régulière et ajouter si seulement si some_computation(x), mais j'aime les compréhensions.

Answer 1

some_computation sera calculé deux fois, pour éviter cela, vous pouvez utiliser filter pour annuler les résultats falsy:

li = list(filter(None, (some_computation(x) for x in y))) 

Ou utiliser une expression generator imbriqué comme Jon Clements suggère:

li = [el for el in (some_computation(x) for x in y) if el] 

Answer 2

1. Est-ce que cela calcule some_computation (x) deux fois que la compréhension de la liste le fait apparaître?

Oui. Il calcule deux fois.

2. Existe-t-il une meilleure façon de procéder? Peut-être de le casser en une boucle normale et d'ajouter si seulement some_computation (x), mais j'aime les compréhensions.

Vous devez faire une boucle régulière. Les générateurs sont à la volée et ne sont pas stockés. Peut-être que vous pouvez prendre l'algorithme pour les x chosing out et faire le calcul

li = [computation(x) for x in y if choose(x)] 

Ou vous avez à faire itération dans la condition if qui équivaut à appeler le calcul deux fois.

Answer 3

1. Oui, il ne some_computation deux fois, essayez et voyez:

   def compute(x): 
       print("Compute...") 
       return 2*x 
   y=[1,3,5] 
   ls=[compute(x) for x in y if compute(x)] 
   

Sortie:

Compute... 
Compute... 
Compute... 
Compute... 
Compute... 
Compute... 

2. Mieux Python chemin:

   ls=[x for x in [compute(z) for z in y] if x]