CODEWITHSUNDEEP
pythonlistdictionarycombinations
grssnbchr
grssnbchr

Reputation: 2974

Combining all combinations of two lists into a dict of special form

I have two lists: 

var_a = [1,2,3,4]

var_b = [6,7]

I want to have a list of dicts as follows:

result = [{'a':1,'b':6},{'a':1,'b':7},{'a':2,'b':6},{'a':2,'b':7},....]

I think the result should be clear.

Upvotes: 0

Views: 2067

Answers (3)

wflynny
wflynny

Reputation: 18521

from itertools import product

a = [1,2,3,4]
b = [6,7]
[dict(zip(('a','b'), (i,j))) for i,j in product(a,b)]

yields

[{'a': 1, 'b': 6},
 {'a': 1, 'b': 7},
 {'a': 2, 'b': 6},
 {'a': 2, 'b': 7},
 {'a': 3, 'b': 6},
 {'a': 3, 'b': 7},
 {'a': 4, 'b': 6},
 {'a': 4, 'b': 7}]

Upvotes: 4

Sukrit Kalra
Sukrit Kalra

Reputation: 34493

If the name of variables is given to you, you could use.

>>> a = [1,2,3,4]
>>> b = [6,7]
>>> from itertools import product
>>> nameTup = ('a', 'b')
>>> [dict(zip(nameTup, elem)) for elem in product(a, b)]
[{'a': 1, 'b': 6}, {'a': 1, 'b': 7}, {'a': 2, 'b': 6}, {'a': 2, 'b': 7}, {'a': 3, 'b': 6}, {'a': 3, 'b': 7}, {'a': 4, 'b': 6}, {'a': 4, 'b': 7}]

Upvotes: 1

inspectorG4dget
inspectorG4dget

Reputation: 113945

[{k:v for k,v in itertools.izip('ab', comb)} for comb in itertools.product([1,2,3,4], [6,7])]

>>> import itertools
>>> [{k:v for k,v in itertools.izip('ab', comb)} for comb in itertools.product([
1,2,3,4], [6,7])]
[{'a': 1, 'b': 6}, {'a': 1, 'b': 7}, {'a': 2, 'b': 6}, {'a': 2, 'b': 7}, {'a': 3
, 'b': 6}, {'a': 3, 'b': 7}, {'a': 4, 'b': 6}, {'a': 4, 'b': 7}]

Upvotes: 5

Related Questions