Блог

Я нахожу исходный код itertools.combinations() функции на python. Это выглядит так.

 def combinations(iterable, r):
pool = tuple(iterable)
n = len(pool)
if r > n:
    return
indices = list(range(r))
print(indices)
yield tuple(pool[i] for i in indices)
while True:
    for i in reversed(range(r)):
        if indices[i] != i   n - r:
            break
    else:
        return
    indices[i]  = 1
    for j in range(i 1, r):
        indices[j] = indices[j-1]   1
    print(indices)
    yield tuple(pool[i] for i in indices)
 

У меня есть такие кортежи:

 pairs = [(0, 1), (0, 2), (0, 3), (1, 2), (1, 3), (2, 3)]
 

Мне нужно сгенерировать четыре пары всех возможных комбинаций, но с условием, что в списке всегда будет только два одинаковых числа. Итак, в этом случае я хочу сгенерировать этот список 3:

 ((0, 1), (0, 2), (1, 3), (2, 3))
((0, 1), (0, 3), (1, 2), (2, 3))
((0, 2), (0, 3), (1, 2), (1, 3))
 

Что мне действительно нужно, так это обновить код комбинаций генерации, потому что в моем реальном приложении мне нужно сгенерировать 23-nty из 80 кортежей. Генерация и фильтрация после этого заняли бы много времени, поэтому мне нужно выявить проблему в части генерации.

Вы можете использовать itertools.combinations , а затем фильтровать результат с помощью collections.Counter :

 from collections import Counter
import itertools as it

pairs = [(0, 1), (0, 2), (0, 3), (1, 2), (1, 3), (2, 3)]
result = filter(
    lambda x: max(Counter(it.chain(*x)).values()) < 3,
    it.combinations(pairs, 4)
)
print(list(result))
 

Вывод:

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