Как получить «дополнительные» комбинации при добавлении элементов в список, для которого я уже получил все комбинации?
Используя Python, я хотел бы вычислить все комбинации из 3 из списка.
Например, list = [a,b,c,d] и комбинации будут такими: [a,b,c], [a,b,d], [a,c,d], [b,c,d] .
И тогда я хотел бы добавить некоторые элементы к исходному списку и получить только дополнительные комбинации из 3.
Например, добавление элементов [e,f] создаст новые комбинации — [a,b,e], [a,b,f], [a,c,e], [a,c,f], [a, d,e], [a,d,f], [a,e,f], [b,c,e], [b,c,f],...
Списки будут большими, поэтому нам нужно избегать двойной генерации комбинаций и последующей фильтрации для получения «дополнительных комбинаций».
Фон: Я использую itertools.combinations, чтобы получить все комбинации (из 3) для списка, содержащего около 100 элементов. Это генерирует множество комбинаций, и я делаю кучу вычислений и тому подобное на основе этих комбинаций, ищу закономерности, совпадения и тому подобное. Я прохожу всю эту обработку, и если у меня нет «успешной» комбинации из 3, я генерирую больше кандидатов для списка (что само по себе занимает много времени). Когда я добавляю в список дополнительных кандидатов (обычно около 10 или около того), я затем перезапускаю анализ комбинаций, что кажется расточительным, поэтому я хотел бы проверять только «дополнительные» комбинации.
Ответы 1
Добавляйте новые элементы один за другим:
from itertools import combinations
old = ['a', 'b', 'c', 'd']
new = ['e', 'f']
for item in new:
for comb in combinations(old, 2):
print(*comb, item)
old.append(item)
Результат (Попробуйте это онлайн!):
a b e
a c e
a d e
b c e
b d e
c d e
a b f
a c f
a d f
a e f
b c f
b d f
b e f
c d f
c e f
d e f
Или, если порядок внутри каждой комбинации не имеет значения (на основе комментария):
from math import comb
from itertools import combinations, islice
old = ['a', 'b', 'c', 'd']
new = ['e', 'f']
new_combs = comb(len(old) + len(new), 3) - comb(len(old), 3)
for comb in islice(combinations(new + old, 3), new_combs):
print(*comb)
Результат (Попробуйте это онлайн!):
e f a
e f b
e f c
e f d
e a b
e a c
e a d
e b c
e b d
e c d
f a b
f a c
f a d
f b c
f b d
f c d
Потрясающе, это было на кончике моего мозга, но я не мог до конца высказать это. Я использовал добавление каждого нового элемента ко всем комбинациям подхода старого списка. Это очень хорошо подходит для моего варианта использования, потому что я могу записать в журнал уже проверенный «старый» список и начать сразу же добавлять новые элементы по мере их создания. Иногда создание новых элементов занимает несколько дней, а затем, по мере роста списка, проверка комбинаций с новым элементом также занимает больше времени. Спасибо!
Есть несколько способов сделать это. Вы можете оставить значение контрольного значения равным последнему набору параметров и просто выполнять итерацию, пока не достигнете контрольного значения. Или вы можете изменить список, по которому вы просматриваете, если совпадение не найдено, как предложено @no-comment.