Cartesian product of a dictionary of lists

by

Ok, thanks @dfan for telling me I was looking in the wrong place. I’ve got it now:

from itertools import product
def my_product(inp):
    return (dict(zip(inp.keys(), values)) for values in product(*inp.values())

EDIT: after years more Python experience, I think a better solution is to accept kwargs rather than a dictionary of inputs; the call style is more analogous to that of the original itertools.product. Also I think writing a generator function, rather than a function that returns a generator expression, makes the code clearer. So:

def product_dict(**kwargs):
    keys = kwargs.keys()
    vals = kwargs.values()
    for instance in itertools.product(*vals):
        yield dict(zip(keys, instance))

and if you need to pass in a dict, list(product_dict(**mydict)). The one notable change using kwargs rather than an arbitrary input class is that it prevents the keys/values from being ordered, at least until Python 3.6.

More Related Contents:

  1. Understanding generators in Python
  2. Generator expressions vs. list comprehensions
  3. Lazy Method for Reading Big File in Python?
  4. Resetting generator object in Python
  5. Generate all permutations of a list without adjacent equal elements
  6. What’s the shortest way to count the number of items in a generator/iterator?
  7. how to split an iterable in constant-size chunks
  8. Is there a generator version of `string.split()` in Python?
  9. Python random sample with a generator / iterable / iterator
  10. How to look ahead one element (peek) in a Python generator?
  11. How do I generate all permutations of a list?
  12. Python generator that groups another iterable into groups of N
  13. Is there an expression for an infinite iterator?
  14. Length of generator output [duplicate]
  15. N-D version of itertools.combinations in numpy
  16. How to get the n next values of a generator in a list (python)
  17. How can I generate all possible IPs from a list of ip ranges in Python?
  18. Best way to receive the ‘return’ value from a python generator
  19. Python: using a recursive algorithm as a generator
  20. all permutations of a binary sequence x bits long
  21. What is the purpose of the “send” function on Python generators?
  22. Return in generator together with yield
  23. What does the “yield” keyword do in Python?
  24. How to clone a Python generator object?
  25. Why do list comprehensions write to the loop variable, but generators don’t? [duplicate]
  26. How can I match up permutations of a long list with a shorter list (according to the length of the shorter list)? [duplicate]
  27. How to use fit_generator with multiple inputs
  28. How to create the union of many sets using a generator expression?
  29. Getting the subsets of a set in Python
  30. How can I count the number of items in an arbitrary iterable (such as a generator)?

Leave a Comment