CODEWITHSUNDEEP
python
Tom
Tom

Reputation: 83

Python combinations with list and items in other lists

I try the code below, is there a efficent way to do this?

c = []
l = [['A1','A2'], ['B1','B2'],  ['C1','C2'] ]

for i in range(0, len(l) - 1):
    for j in range(i+1, len(l)): 
        c.append(sorted([l[i][0],l[i][1],l[j][0]]))
        c.append(sorted([l[i][0],l[i][1],l[j][1]]))
        c.append(sorted([l[i][0],l[j][0],l[j][1]]))
        c.append(sorted([l[i][1],l[j][0],l[j][1]]))

print(c)

Out put:

[['A1', 'A2', 'B1'], ['A1', 'A2', 'B2'], ['A1', 'B1', 'B2'],
['A2', 'B1', 'B2'], ['A1', 'A2', 'C1'], ['A1', 'A2', 'C2'], 
['A1', 'C1', 'C2'], ['A2', 'C1', 'C2'], ['B1', 'B2', 'C1'], 
['B1', 'B2', 'C2'], ['B1', 'C1', 'C2'], ['B2', 'C1', 'C2']

Upvotes: 0

Views: 83

Answers (2)

Sardorbek Imomaliev
Sardorbek Imomaliev

Reputation: 15400

Or in one line

from itertools import product

c = [[k] + i for i, j in product(l, l) if j!=i for k in j]

Upvotes: 1

Pan Long
Pan Long

Reputation: 1032

Try this:

# group every 2 lists in list l
ll = list(itertools.combinations(l, 2))

# generate all combinations of 3 elements out from each 2 lists
c = [list(itertools.combinations(a + b, 3)) for (a, b) in ll]

# concate all elements
c = sum(c, [])

Upvotes: 2

Related Questions