0

getting simple permutations of a string is easy but how do you do if you want different length permutations For eg:

def permutations(string):
    if len(string) == 1:
        return [string]
    else:
        perms = []
        for i in range(len(string)):
            for perm in permutations(string[:i] + string[i+1:]):
                perms.append(string[i] + perm)
    return perms
permutations('abc')

returns ['abc', 'acb', 'bac', 'bca', 'cab', 'cba']

but i want ['abc', 'acb', 'bac', 'bca', 'cab', 'cba','ab','ba','bc','cb','ca','ac','a','b','c']

How do i get that extra arrangements????

VPfB
  • 14,927
  • 6
  • 41
  • 75

2 Answers2

1

you should use permutations from itertools

from itertools import permutations
all_perm = []
st="abc"
for i in range(1,len(st)+1):
    all_perm += list(permutations(st,i))
    
all_perms_str = [''.join(str(e) for e in perm) for perm in all_perm]
print (all_perms_str)

output:

['a', 'b', 'c', 'ab', 'ac', 'ba', 'bc', 'ca', 'cb', 'abc', 'acb', 'bac', 'bca', 'cab', 'cba']
Ohad Sharet
  • 1,097
  • 9
  • 15
0

With just one line added:

def permutations(string):
    if len(string) == 1:
        return [string]
    else:   # note: else not needed after return
        perms = []
        for i in range(len(string)):
            for perm in permutations(string[:i] + string[i+1:]):
                perms.append(string[i] + perm)
            perms.append(string[i])  # <--- new line
    return perms

Output:

['abc', 'ab', 'acb', 'ac', 'a', 'bac', 'ba', 'bca', 'bc', 'b', 'cab', 'ca', 'cba', 'cb', 'c']
VPfB
  • 14,927
  • 6
  • 41
  • 75