Exécuter fonction sur toutes les combinaisons possibles de paramètres

Question

J'ai ensembles de valeurs que je veux appliquer en tant que paramètres à une fonction:

params = { 
    'a': [1, 2, 3], 
    'b': [5, 6, 7], 
    'x': [None, 'eleven', 'f'], 
    # et cetera 
} 

Je veux courir myfunc() avec toutes les combinaisons possibles, donc myfunc(a=1, b=5, x=None ...), myfunc(a=2, b=5, x=None ...) .. myfunc(a=3, b=7, x='f' ...). Y at-il quelque chose (par exemple dans itertools) qui peut aider? J'ai pensé utiliser itertools.product() mais cela ne garde pas les noms des paramètres et me donne juste des tuples des combinaisons.

Answer 1

Vous pouvez utiliser itertools.product pour obtenir toutes les combinaisons d'arguments:

>>> import itertools 
>>> for xs in itertools.product([1,2], [5,6], ['eleven', 'f']): 
...  print(xs) 
... 
(1, 5, 'eleven') 
(1, 5, 'f') 
(1, 6, 'eleven') 
(1, 6, 'f') 
(2, 5, 'eleven') 
(2, 5, 'f') 
(2, 6, 'eleven') 
(2, 6, 'f') 

Avec Argument list unpacking, vous pouvez appeler myfunc avec toutes les combinaisons d'arguments mot-clé:

params = { 
    'a': [1, 2, 3], 
    'b': [5, 6, 7], 
    'x': [None, 'eleven', 'f'], 
} 

def myfunc(**args): 
    print(args) 

import itertools 
keys = list(params) 
for values in itertools.product(*map(params.get, keys)): 
    myfunc(**dict(zip(keys, values))) 

sortie:

{'a': 1, 'x': None, 'b': 5} 
{'a': 1, 'x': None, 'b': 6} 
{'a': 1, 'x': None, 'b': 7} 
{'a': 1, 'x': 'eleven', 'b': 5} 
{'a': 1, 'x': 'eleven', 'b': 6} 
{'a': 1, 'x': 'eleven', 'b': 7} 
{'a': 1, 'x': 'f', 'b': 5} 
... 

Answer 2

Commande de .keys et .values sont garantis dans toutes les versions de Python (sauf si dict est modifié qui ne se fait pas ici), alors cela pourrait être un peu trivial:

from itertools import product 

for vals in product(*params.values()): 
    myfunc(**dict(zip(params, vals))) 

Vous pouvez trouver le gurantee dans le docs:

Si les clés, les valeurs et les vues d'éléments sont itérées sans modification du dictionnaire, l'ordre des éléments correspondra directement à .

---

Demo:

for vals in product(*params.values()): 
    print(dict(zip(params, vals))) 

---

{'a': 1, 'x': None, 'b': 5} 
{'a': 1, 'x': None, 'b': 6} 
{'a': 1, 'x': None, 'b': 7} 
{'a': 1, 'x': 'eleven', 'b': 5} 
{'a': 1, 'x': 'eleven', 'b': 6} 
{'a': 1, 'x': 'eleven', 'b': 7} 
{'a': 1, 'x': 'f', 'b': 5} 
{'a': 1, 'x': 'f', 'b': 6} 
{'a': 1, 'x': 'f', 'b': 7} 
...