Как создавать массивы с комбинациями между определенными индексами фиксированной длины и фиксированной суммы

Например:

array = [4,3,2,0,0,0,0,0,0]

0-й индекс должен иметь только комбинации с 3-м индексом и 6-м индексом. 1-й индекс должен иметь только комбинации с 4-м индексом и 7-м индексом. 2-й индекс должен иметь только комбинации с 5-м индексом и 8-м индексом. (сумма должна оставаться неизменной между этими индексами). Тогда вывод должен быть:

[1,2,2,1,1,0,2,0,0]
[2,1,1,1,1,1,1,1,0]...

В обеих этих комбинациях сумма между соответствующими индексами (перечисленными выше) остается неизменной.

python

14.05.2022 11:35

Почему в Python есть оператор "pass"?

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

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

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

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Используя функцию findPairs, полученную из ответ на ваш предыдущий вопрос:

from itertools import product
  
def findPairs(sum_value, len_value):
    lst = range(sum_value + 1)

    return [
        pair 
        for pair in product(lst, repeat=len_value) 
        if sum(pair) == sum_value
    ]

import itertools

combinations = itertools.product(findPairs(array[0], 3), findPairs(array[1], 3), findPairs(array[2], 3))
result = [list(itertools.chain(*zip(p1, p2, p3))) for p1, p2, p3 in combinations]
print(result[0:10])

[[0, 0, 0, 0, 0, 0, 4, 3, 2], [0, 0, 0, 0, 0, 1, 4, 3, 1], 
[0, 0, 0, 0, 0, 2, 4, 3, 0], [0, 0, 1, 0, 0, 0, 4, 3, 1], 
[0, 0, 1, 0, 0, 1, 4, 3, 0], [0, 0, 2, 0, 0, 0, 4, 3, 0], 
[0, 0, 0, 0, 1, 0, 4, 2, 2], [0, 0, 0, 0, 1, 1, 4, 2, 1], 
[0, 0, 0, 0, 1, 2, 4, 2, 0], [0, 0, 1, 0, 1, 0, 4, 2, 1]]

...

14.05.2022 12:17

Другие вопросы по теме

Ошибка импорта: «SimpleExperiment» при запуске примера кода BOTORCH

Как указать tox использовать доверенный хост и индекс?

Как прочитать файл csv, отредактированный из excel в python?

Преобразование заданной строки временной метки UTC в временную метку unix

Как удалить английские, но не английские слова в Python?

Как строить пары на разных участках с разницей на стороне

Получение ошибки при извлечении данных из базы данных (mysql). n="+".join(n) TypeError: может присоединиться только к итерируемому

Как получить один CSV-файл после применения partitionBy в Pysark

Как написать простой код для вычисления с плавающей запятой в Python?

Любой короткий способ создать комбинацию массивов с фиксированной длиной и суммой?

Как создавать массивы с комбинациями между определенными индексами фиксированной длины и фиксированной суммы

Ответы 1

Другие вопросы по теме

Похожие вопросы