Каков наилучший способ создать все возможные трехбуквенные строки? - PullRequest
33 голосов
/ 16 августа 2011

Я генерирую все возможные трехбуквенные ключевые слова e.g. aaa, aab, aac.... zzy, zzz ниже мой код:

alphabets = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']

keywords = []
for alpha1 in alphabets:
    for alpha2 in alphabets:
        for alpha3 in alphabets:
            keywords.append(alpha1+alpha2+alpha3)

Можно ли добиться этой функциональности более гладким и эффективным способом?

Ответы [ 7 ]

79 голосов
/ 16 августа 2011
keywords = itertools.product(alphabets, repeat = 3)

См. Документацию для itertools.product.Если вам нужен список строк, просто используйте

keywords = [''.join(i) for i in itertools.product(alphabets, repeat = 3)]

alphabets и не обязательно должен быть списком, это может быть просто строка, например:

from itertools import product
from string import ascii_lowercase
keywords = [''.join(i) for i in product(ascii_lowercase, repeat = 3)]

будет работать, если вам нужны строчные буквы ascii .

15 голосов
/ 16 августа 2011

Вы также можете использовать карту вместо понимания списка (это один из случаев, когда карта все еще быстрее, чем LC)

>>> from itertools import product
>>> from string import ascii_lowercase
>>> keywords = map(''.join, product(ascii_lowercase, repeat=3))

Этот вариант понимания списка также быстрее, чем при использовании ''.join

>>> keywords = [a+b+c for a,b,c in product(ascii_lowercase, repeat=3)]
4 голосов
/ 19 августа 2011

Вы также можете сделать это без каких-либо внешних модулей, выполнив простой расчет.
PermutationIterator - это то, что вы ищете.

def permutation_atindex(_int, _set, length):
    """
    Return the permutation at index '_int' for itemgetter '_set'
    with length 'length'.
    """
    items = []
    strLength = len(_set)
    index = _int % strLength
    items.append(_set[index])

    for n in xrange(1,length, 1):
        _int //= strLength
        index = _int % strLength
        items.append(_set[index])

    return items

class PermutationIterator:
    """
    A class that can iterate over possible permuations
    of the given 'iterable' and 'length' argument.
    """

    def __init__(self, iterable, length):
        self.length = length
        self.current = 0
        self.max = len(iterable) ** length
        self.iterable = iterable

    def __iter__(self):
        return self

    def __next__(self):
        if self.current >= self.max:
            raise StopIteration

        try:
            return permutation_atindex(self.current, self.iterable, self.length)
        finally:
            self.current   += 1

Дайте ему итеративный объект и целое число в качестве длины вывода.

from string import ascii_lowercase

for e in PermutationIterator(ascii_lowercase, 3):
    print "".join(e)

Это начнется с 'aaa' и закончится 'zzz'.

4 голосов
/ 16 августа 2011
from itertools import combinations_with_replacement

alphabets = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']

for (a,b,c) in combinations_with_replacement(alphabets, 3):
    print a+b+c
2 голосов
/ 16 августа 2011
chars = range(ord('a'), ord('z')+1);
print [chr(a) + chr(b) +chr(c) for a in chars for b in chars for c in chars]
0 голосов
/ 28 февраля 2019

Мы могли бы решить эту проблему без использования инструментов itertools, используя два определения функций:

def combos(alphas, k):
    l = len(alphas)
    kRecur(alphas, "", l, k)

def KRecur(alphas, prfx, l, k):
    if k==0:
        print(prfx)
    else:
        for i in range(l):
            newPrfx = prfx + alphas[i]
            KRecur(alphas, newPrfx, l, k-1)

Это делается с помощью двух функций, чтобы избежать сброса длины альфа, а вторая функция выполняет итерацию самостоятельно, пока не достигнетak 0, чтобы вернуть k-mer для этого цикла i.

Принята из решения Абхинав Рамана на Geeks4Geeks

0 голосов
/ 21 января 2017
print([a+b+c for a in alphabets for b in alphabets for c in alphabets if a !=b and b!=c and c!= a])

Удаляет повторение символов в одной строке

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...