How to create a set of values from multiple list in Python

Question

I have below data in config.json:

{
  "data1": [7, 8],
  "data2": [2015, 2016],
  "data3": [5, 10],
}

It has 3 list with length as 2 but it can have multiple list with n length. I have to read this data and then create a list of list which should have all the values just like set. So output should look like below:

[
    [7, 2015, 5],
    [7, 2015, 10],
    [7, 2016, 5],
    [7, 2016, 10],
    [8, 2015, 5],
    [8, 2015, 10],
    [8, 2016, 5],
    [8, 2016, 10]
]


json_data = open("config.json")
config = json.load(json_data)
json_data.close()

data_list = []
for be in config["data1"]:
    for eof in config["data2"]:
        for bd in config["data3"]:
            data_list.append(bd)

How can I modify the data to have output like above?

I am a little confused, will the number of lists be dynamic and you want to find every combination of those? Is the length ([7,8]) of 2 items static, or is it also dynamic? — Zaid Al Shattle
– Zaid Al Shattle, Commented Oct 4, 2021 at 11:33
@ZaidAlShattle length of list is dynamic but I can say number of list can be static — S Andrew
– S Andrew, Commented Oct 4, 2021 at 11:39

Mahdi F. · Accepted Answer · 2022-06-05 14:56:39Z

4

You can use itertools.product:

list(it.product(*data.values()))

edited Jun 5, 2022 at 14:56

Mahdi F.

24.1k5 gold badges25 silver badges32 bronze badges

answered Oct 4, 2021 at 11:39

a_guest

36.7k15 gold badges75 silver badges137 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

S Andrew Over a year ago

Do you mean I have to run this in a for loop where data will be the data from config file

a_guest Over a year ago

I assumed data is the example dictionary you provided in your question.

gimix · Accepted Answer · 2021-10-04 11:40:35Z

2

Use itertools.product which gives the Cartesian product of a series of iterables:

from itertools import product
data = {'data1': [7, 8], 'data2': [2015, 2016], 'data3': [5, 10]}
cprod = product(*data.values())
for cp in cprod:
    print(cprod)
(7, 2015, 5)
(7, 2015, 10)
(7, 2016, 5)
(7, 2016, 10)
(8, 2015, 5)
(8, 2015, 10)
(8, 2016, 5)
(8, 2016, 10)

answered Oct 4, 2021 at 11:40

gimix

3,8432 gold badges8 silver badges27 bronze badges

Comments

NiziL · Accepted Answer · 2021-10-04 11:59:40Z

Here's some information about a_guest's answer, as you seems to have some trouble understand it.

json_data = open("config.json")
config = json.load(json_data)
json_data.close()

cproduct = list(itertools.product(*config.values()))

product function will give you the cartesian product, that's exactly what you need : all possible combination of elements from different set. We basically use it as product(set_1, set_2, ..., set_n).

Here, set_1 should be the list stored with the "data1" key, set_2 with "data2" and so on. So you could basically write it.product(config['data1'], config['data2'], config['data3']).

But how can we deals with new keys in the json without changing our code ?

To achieve this, guest_a relies on two things :

config.values() which return a list of every values, so in your case a list containing all the list defined in your json. This way, you'll get the new list if you add a new "data4" key into your json file.
Unpacking * operator, which allow us to use any iterable as a list of arguments.

So basically, it.product(*config.values()) is equivalent to it.product(config["data1"], ..., config["data_n"]), but is much more flexible.

ashhad ullah · Accepted Answer · 2021-10-04 11:42:47Z

1

you have done all the work just add

    data_list = []
    for be in config["data1"]:
        for eof in config["data2"]:
            for bd in config["data3"]:
                data_list.append([be,eof,bd])

answered Oct 4, 2021 at 11:42

ashhad ullah

1165 bronze badges

Collectives™ on Stack Overflow

How to create a set of values from multiple list in Python

4 Answers 4

2 Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

2 Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related