Pythonic способ подражать itertools.product в Python 2.4 - PullRequest
3 голосов
/ 12 октября 2011

У меня есть скрипт Python 3, который использует itertools.product, но мне нужно иметь возможность запускать его на машине, на которой установлен только Python 2.4.Поскольку itertools.product является новым в Python 2.6, у меня больше нет доступа к этой функции.

Как я могу эмулировать itertools.product в Python 2.4 питонским способом?

Ответы [ 2 ]

6 голосов
/ 12 октября 2011

Эквивалентный код от http://docs.python.org/library/itertools.html#itertools.product

def product(*args, **kwds):
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
    pools = map(tuple, args) * kwds.get('repeat', 1)
    result = [[]]
    for pool in pools:
        result = [x+[y] for x in result for y in pool]
    for prod in result:
        yield tuple(prod)
5 голосов
/ 12 октября 2011

Я не слишком знаком с Python 2.4, но за 2,7 документа :

Эта функция эквивалентна следующему коду, за исключением того, что фактическая реализация не создает промежуточных результатов в память:

def product(*args, **kwds):
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
    pools = map(tuple, args) * kwds.get('repeat', 1)
    result = [[]]
    for pool in pools:
        result = [x+[y] for x in result for y in pool]
    for prod in result:
        yield tuple(prod)
...