Блог

Я пытаюсь сгенерировать комбинации первого элемента списка со всеми остальными элементами. Выполнение этого с небольшим списком не является проблемой, если сначала получить все возможные комбинации, а затем отбросить те, которые мне не нужны. Пример:

 import numpy as np
import itertools

lst = ['a','b','c','d','e','f']
element_of_interest = 'a'

all_combinations = np.array(list(itertools.combinations(lst, 3)))

idx = np.where(all_combinations == element_of_interest)
combinations_of_interest = all_combinations[idx[0]]

print(combinations_of_interest)

>>> [['a' 'b' 'c']
 ['a' 'b' 'd']
 ['a' 'b' 'e']
 ['a' 'b' 'f']
 ['a' 'c' 'd']
 ['a' 'c' 'e']
 ['a' 'c' 'f']
 ['a' 'd' 'e']
 ['a' 'd' 'f']
 ['a' 'e' 'f']]
 

Но в моем приложении (список из 290 элементов, r = 4) этот способ недостаточно хорош, так как он очень неэффективен.

Кто-нибудь может придумать лучший способ сделать это?

Выньте элемент из списка, а затем добавьте его позже ко всем остальным комбинациям:

 In [9]: import itertools
   ...:
   ...: lst = ['a','b','c','d','e','f']
   ...: element_of_interest = 'a'

In [11]: out = [['a']   list(i) for i in itertools.combinations(lst[1:], 2)]

In [12]: out
Out[12]:
[['a', 'b', 'c'],
 ['a', 'b', 'd'],
 ['a', 'b', 'e'],
 ['a', 'b', 'f'],
 ['a', 'c', 'd'],
 ['a', 'c', 'e'],
 ['a', 'c', 'f'],
 ['a', 'd', 'e'],
 ['a', 'd', 'f'],
 ['a', 'e', 'f']]