Making a product of an items in dictionaries

Question

let's say I have this list of states:

states1 = ['one', 'two', 'four']
states2 = ['far away', 'normal', 'to close']
states3 = ['thor', 'iron man']

and I want to make a product of them

import itertools

state_indexes = [s for s in itertools.product(states1, states2, states3)]
print(state_indexes)

"""
[('one', 'far away', 'thor'), ('one', 'far away', 'iron man'), ('one', 'normal', 'thor'), ('one', 'normal', 'iron man'), ('one', 'to close', 'thor'), ('one', 'to close', 'iron man'), ('two', 'far away', 'thor'), ('two', 'far away', 'iron man'), ('two', 'normal', 'thor'), ('two', 'normal', 'iron man'), ('two', 'to close', 'thor'), ('two', 'to close', 'iron man'), ('four', 'far away', 'thor'), ('four', 'far away', 'iron man'), ('four', 'normal', 'thor'), ('four', 'normal', 'iron man'), ('four', 'to close', 'thor'), ('four', 'to close', 'iron 
man')]
"""

but this is hard-coded way of doing so, I want to create the list of states automatically so I created this dictionary:

my_dictionary = {}

for i in range(10):
    my_dictionary[f"item_{i}"] = ['state1', 'state2']

"""
{'item_0': ['state1', 'state2'], 'item_1': ['state1', 'state2'], 'item_2': ['state1', 'state2'], 'item_3': ['state1', 'state2'], 'item_4': ['state1', 'state2'], 'item_5': ['state1', 'state2'], 'item_6': ['state1', 'state2'], 'item_7': ['state1', 'state2'], 'item_8': ['state1', 'state2'], 'item_9': ['state1', 'state2']}
"""

Until now it works, but how can I recreate the state_indexes list?

This is one of the approaches I tried:

state_indexes = [s for s in itertools.product(my_dictionary)]
print(state_indexes)
"""
[('item_0',), ('item_1',), ('item_2',), ('item_3',), ('item_4',), ('item_5',), ('item_6',), ('item_7',), ('item_8',), ('item_9',)]
"""

How can I fix that? Thank you very much

EDIT: Expected output is basically this (as stated in the example above):

"""
[('one', 'far away', 'thor'), ('one', 'far away', 'iron man'), ('one', 'normal', 'thor'), ('one', 'normal', 'iron man'), ('one', 'to close', 'thor'), ('one', 'to close', 'iron man'), ('two', 'far away', 'thor'), ('two', 'far away', 'iron man'), ('two', 'normal', 'thor'), ('two', 'normal', 'iron man'), ('two', 'to close', 'thor'), ('two', 'to close', 'iron man'), ('four', 'far away', 'thor'), ('four', 'far away', 'iron man'), ('four', 'normal', 'thor'), ('four', 'normal', 'iron man'), ('four', 'to close', 'thor'), ('four', 'to close', 'iron 
man')]
"""

So it would look like this (I know that it is not very intuitive, that's why I used the example)

"""
[('state1', 'state1', 'state1',...,'state1'), ('state1', 'state1',..., 'state2'),..., ('state2', 'state2',... ,'state2')]
"""

Does this answer your question? [Get the cartesian product of a series of lists?](https://stackoverflow.com/questions/533905/get-the-cartesian-product-of-a-series-of-lists) — Harun Yilmaz, Apr 16 '20 at 08:52
It is not clear if you want combinations over the keys or the values in the dict? — user2390182, Apr 16 '20 at 08:54

user2390182 · Accepted Answer · 2020-04-16T08:58:57.633

2

For the product of all the lists of states, you would have to unpack the dict's values, using the * operator:

state_indexes = [s for s in itertools.product(*my_dictionary.values())]
# state_indexes = list(itertools.product(*my_dictionary.values()))

Note, however, that the order of the dict values is not guaranteed in all Python versions. You might have to use a list or an collections.OrderedDict.

edited Apr 16 '20 at 08:58

answered Apr 16 '20 at 08:56

user2390182

72,016
6
67
89

Making a product of an items in dictionaries

1 Answers1