Itertools.product изменить порядок генерации

@cyberturist · Регистрация: 01.12.2018

Студворк — интернет-сервис помощи студентам

Здравствуйте,
Суть, проблемы описана в следующем посте, только в моем случае будут исключительно числа.
То есть генерация комбинаций должна происходить от наименьшей суммы всех членов к наибольшей. Например,по аналогии с itertools.product, генератор my_product(range(3), repeat=3) должен давать: (0,0,0), (0,0,1),(0,1,0),..,(0,1,1),..,(0,0,2),.. .(2,0,0) и того 27 комбинаций. itertools.product(range(3), repeat=3) дает те же комбинации, но к сожалению, их генерация происходит в другом порядке. Нужного порядка генерации можно добиться методом предложенным в вышеуказанном посте с сортировкой

Python
1
2
def my_product(*args):
    return list(sorted(itertools.product(*args), key=lambda x: (sum(x), x)))

но тогда съедается вся память.
Мне это нужно, чтобы при определенной сумме чисел, которая будет больше пороговой, остановить генерацию. Например, сейчас работает так:

Python
1
2
for i in product(range(excitation + 1), repeat=len(arr[0])):
  if sum(i) <= excitation + 1:

но так как перебирается много не нужных комбинаций, то тратится много времени. Смотрю реализацию функций itertools и ничего рабочего не могу придумать. Пожалуйста, подскажите.

@vrm2 · 23.10.2019, 10:34

Python
1
2
3
4
5
6
7
8
9
10
11
12
def split(n, cnt):
    if cnt <= 1:
        yield [n]
        return
    for x in range(n+1):
        for xs in split(n-x, cnt-1):
            yield [x] + xs
 
def gen(max_sum, cnt):
    assert max_sum >= 0
    for n in range(max_sum+1):
        yield from split(n, cnt)

Результат:

Code
1
2
3
4
5
6
7
8
9
10
11
12
list(gen(max_sum=2, cnt=3))
 
[[0, 0, 0],
 [0, 0, 1],
 [0, 1, 0],
 [1, 0, 0],
 [0, 0, 2],
 [0, 1, 1],
 [0, 2, 0],
 [1, 0, 1],
 [1, 1, 0],
 [2, 0, 0]]

Новые блоги и статьи Все статьи Все блоги /
Подстановка значения реквизита справочника в табличную часть документа Maks 10.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "ПланированиеПерсонала", разработанного в конфигурации КА2. Задача: при выборе сотрудника (справочник Сотрудники) в ТЧ документа. . .	Очистка реквизитов документа при копировании Maks 09.04.2026 Алгоритм из решения ниже применим как для типовых, так и для нетиповых документов на самых различных конфигурациях. Задача: при копировании документа очищать определенные реквизиты и табличную. . .	модель ЗдравоСохранения 8. Подготовка к разному выполнению заданий anaschu 08.04.2026 https:/ / github. com/ shumilovas/ med2. git main ветка * содержимое блока дэлэй из старой модели теперь внутри зайца новой модели 8ATzM_2aurI	Блокировка документа от изменений, если он открыт у другого пользователя Maks 08.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа, разработанного в конфигурации КА2. Задача: запретить редактирование документа, если он открыт у другого пользователя. / / . . .
Система безопасности+живучести для сервера-слоя интернета (сети). Двойная привязка. Hrethgir 08.04.2026 Далее были размышления о системе безопасности. Сообщения с наклонным текстом - мои. А как нам будет можно проверить, что ссылка наша, а не подделана хулиганами, которая выбросит на другую ветку и. . .	Модель ЗдрввоСохранения 7: больше работников, больше ресурсов. anaschu 08.04.2026 работников и заданий может быть сколько угодно, но настроено всё так, что используется пока что только 20% kYBz3eJf3jQ	Дальние перспективы сервера - слоя сети с космологическим дизайном интефейса карты и логики. Hrethgir 07.04.2026 Дальнейшее ближайшее планирование вывело к размышлениям над дальними перспективами. И вот тут может быть даже будут нужны оценки специалистов, так как в дальних перспективах всё может очень сильно. . .	Горе от ума kumehtar 07.04.2026 Эта мне ментальная установка, что вот прямо сейчас, мол, мне для полного счастья не хватает (нужное вписать), и когда я этого достигну - тогда и полный кайф. Одна из самых сильных ловушек на пути. . . .

Itertools.product изменить порядок генерации

Решение