Генерация всех перестановок с отдельными ограничениями для каждого индекса

Question

Генерация всех перестановок с отдельными ограничениями для каждого индекса

1

Скажем, у нас есть список, например, [3, 2, 1]. Я хотел бы сгенерировать все перестановки этого списка в форме:

[1, 1, 1], [2, 1, 1], [3, 1, 1], [1, 2, 1], [2, 2, 1] , [3, 2, 1]

для любого списка длины n. Таким образом, значение i го элемента исходного списка является верхним пределом для значения i го элемента всех перестановок.

Я также хотел бы использовать генератор, используя yield, так как список ввода может быть довольно большим (например, n = 30).

До сих пор я использовал что-то вроде этого:

itertools.product(range(1, 5), repeat=5)

Который имеет следующий выход при использовании в цикле for:

(1, 1, 1, 1, 1), (1, 1, 1, 1, 2), (1, 1, 1, 1, 3), (1, 1, 1, 1, 4), (1, 1, 1, 2, 1), (1, 1, 1, 2, 2), (1, 1, 1, 2, 3), ...

Однако я не думаю, что он позволяет указать пользовательские лимиты для каждого элемента перестановок.

Также обратите внимание, что элементы входного списка необязательно должны быть последовательными числами, поэтому [25, 17, 10, 4] является допустимым вводом.

aL_eX 27 июль 2018, в 00:32

Источник

0

Почему [3,3,3] возможности?
rafaelc 26 июль 2018, в 22:03
0

@RafaelC Поскольку вторым элементом является 2, второй элемент всех перестановок должен быть меньше или равен 2. Я отредактировал вопрос, чтобы прояснить это.
aL_eX 26 июль 2018, в 22:05
0

Имеет ли значение порядок? Есть [(1, 1, 1), (1, 2, 1), (2, 1, 1), (2, 2, 1), (3, 1, 1), (3, 2, 1)] правильное решение?
Francisco Couzo 26 июль 2018, в 22:08
0

@FranciscoCouzo Порядок вывода в вопросе был бы предпочтительнее, так как они являются показателями, используемыми где-то еще, и я хотел бы, чтобы самые большие значения сначала отображались слева от списка.
aL_eX 26 июль 2018, в 22:10

Показать ещё 2 комментария

Теги:

python

list

generator

permutation

3 ответа

1

Вы можете фильтровать результаты itertools.product

>>> from itertools import product
>>> l = [3,2,1]
>>> list(filter(lambda t: all(x<=y for x,y in zip(t,l)), product(l, repeat=len(l))))
[(3, 2, 1), (3, 1, 1), (2, 2, 1), (2, 1, 1), (1, 2, 1), (1, 1, 1)]

Sunitha 26 июль 2018, в 19:44

0

Не могли бы вы указать , какие l в примере?
aL_eX 26 июль 2018, в 22:17
1

@aL_eX Извините .. Обновлено
Sunitha 26 июль 2018, в 22:21
1

Это действительно неэффективно, оно генерирует МНОГО кортежей, которые просто отбрасываются
Francisco Couzo 26 июль 2018, в 22:25
0

Кроме того, я думаю, что repeat=3 следует заменить на repeat=len(l) для общего случая.
aL_eX 26 июль 2018, в 22:27
1

@aL_eX. Благодарю. обновленный
Sunitha 26 июль 2018, в 22:29

Показать ещё 3 комментария

0

Вы ищете декартово произведение множества диапазонов, аргумент которых определяется вашим входным списком:

from itertools import product
lims = [3, 2, 1]
gen = product(*(range(1,lim+1) for lim in lims))
print(list(gen))

В результате

[(1, 1, 1), (1, 2, 1), (2, 1, 1), (2, 2, 1), (3, 1, 1), (3, 2, 1)]

Andras Deak 26 июль 2018, в 21:02

Ещё вопросы

@RafaelC Поскольку вторым элементом является 2, второй элемент всех перестановок должен быть меньше или равен 2. Я отредактировал вопрос, чтобы прояснить это.
Имеет ли значение порядок? Есть [(1, 1, 1), (1, 2, 1), (2, 1, 1), (2, 2, 1), (3, 1, 1), (3, 2, 1)] правильное решение?
@FranciscoCouzo Порядок вывода в вопросе был бы предпочтительнее, так как они являются показателями, используемыми где-то еще, и я хотел бы, чтобы самые большие значения сначала отображались слева от списка.
Не могли бы вы указать , какие l в примере?
Это действительно неэффективно, оно генерирует МНОГО кортежей, которые просто отбрасываются
Кроме того, я думаю, что repeat=3 следует заменить на repeat=len(l) для общего случая.

Francisco Couzo · Accepted Answer · 2018-07-26T19-33-00.000Z

Эта рекурсивная функция возвращает генератор в желаемом порядке:

def f(limits):
    if not limits:
        yield ()
        return

    for l in f(limits[1:]):
        for i in range(1, limits[0]+1):
            yield (i,) + l

>>> print(list(f([3, 2, 1])))
[(1, 1, 1), (2, 1, 1), (3, 1, 1), (1, 2, 1), (2, 2, 1), (3, 2, 1)]