Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

combinations with two elements

With python. It's possible to permute elements in a list with this method:

def perms(seq):
    if len(seq) <= 1:
        perms = [seq]
    else:
        perms = []
        for i in range(len(seq)):
            sub = perms(seq[:i]+seq[i+1:]) 
            for p in sub:    
                perms.append(seq[i:i+1]+p)

return perms

if a list is: seq = ['A', 'B', 'C'], the result will be..

[['A', 'B', 'C'], ['A', 'C', 'B'], ['B', 'A', 'C'], ['B', 'C', 'A'], ['C', 'A', 'B'], ['C', 'B', 'A']]

HOW TO modify this method, to make permutations two terms a time? I mean, if the list is: seq = ['A', 'B', 'C']. I wanna receive [['A', 'B'], ['A', 'C'], ['B', 'C'].

I can't do it. I'm trying but I can't. Thanks for the help.

like image 319
user3104548 Avatar asked Dec 24 '13 14:12

user3104548


2 Answers

Consider using the combinations function in Python's itertools module:

>>> list(itertools.combinations('ABC', 2))
[('A', 'B'), ('A', 'C'), ('B', 'C')]

This does what it says: give me all combinations with two elements from the sequence 'ABC'.

like image 67
Simeon Visser Avatar answered Sep 18 '22 17:09

Simeon Visser


def getCombinations(seq):
    combinations = list()
    for i in range(0,len(seq)):
        for j in range(i+1,len(seq)):
            combinations.append([seq[i],seq[j]])
    return combinations

>>> print(getCombinations(['A','B','C']))
[['A', 'B'], ['A', 'C'], ['B', 'C']]
like image 45
Adam Smith Avatar answered Sep 17 '22 17:09

Adam Smith