Comment aplatir le rendement lors de la production d'un générateur?

Question

J'ai cette fonction g() mon but est de produire un résultat similaire à product() mais avec un nombre d'apparence contraint pour chaque groupe de caractères. Donc, dans le code ci-dessous au lieu de faire product(a, repeat=3)g() est censé donner toutes les combinaisons où les caractères de a apparaissent 2 fois et les caractères de b apparaissent 1 fois. Voici ce que j'ai:

from itertools import permutations, product 

a='ABC' 
b=['*', '%3A'] 
l=[a]*2+[b] 

def g(list): 
    for p in permutations(list): 
     yield product(*p) 


result=g(l) 

Le problème que je suis face est que lorsque je yield le résultat que je suis un itérateur imbriqué qui est pénible à utiliser et quand je l'utilise retourner retourne juste le générateur correspondant à seulement le premier tour de boucle, comme si je l'ai fait: g() seulement composé de def g(): return product(*next(permutations(list))).

Answer 1

Si vous êtes python 3.3 ou une version ultérieure, vous pouvez utiliser PEP 380: Syntax for Delegating to a Subgenerator ... Il suffit de faire yield from

from itertools import permutations, product 

a='ABC' 
b=['*', '%3A'] 
l=[a]*2+[b] 

def g(list): 
    for p in permutations(list): 
     yield from product(*p) 


result=g(l)