Питонический способ генерации пар

Question

Я хочу что-то вроде приведенного ниже кода, но в "питоническом" стиле или с использованием стандартной библиотеки:

def combinations(a,b):
    for i in a:
        for j in b:
             yield(i,j)

Answer 1

Это не совсем «комбинации» в смысле комбинаторики, это скорее элементы из декартовых произведений a и b.Функция в стандартной библиотеке для генерации этих пар: itertools.product():

for i, j in itertools.product(a, b):
    # whatever

Answer 2

Как сказал @Sven, ваш код пытается получить все упорядоченные пары элементов списков a и b.В этом случае itertools.product(a,b) это то, что вы хотите.Если вместо этого вы на самом деле хотите «комбинации», которые являются неупорядоченными парами отдельных элементов списка a, то вы хотите itertools.combinations(a,2).

>>> for pair in itertools.combinations([1,2,3,4],2):
...    print pair
...
(1, 2)
(1, 3)
(1, 4)
(2, 3)
(2, 4)
(3, 4)

Answer 3

Также будет работать вложенное выражение генератора:

product = ((i, j) for i in a for j in b)
for i, j in product:
    # ...

Answer 4

Библиотека itertools имеет функции комбинаторики. Как сказал Свен, itertools.product будет подходящей функцией в этом случае:

list(itertools.product('ab', 'cd'))
[('a', 'c'), ('a', 'd'), ('b', 'c'), ('b', 'd')]

Answer 5

>>>a=[1,2,3]
>>>b=[4,5,6]
>>>zip(a,b)
[(1, 4), (2, 5), (3, 6)] 

Answer 6

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

Если вы хотите, чтобы все неупорядоченные пары (то есть (1, 2) и (2, 1) считались одной и той же парой), вам необходимо отфильтровать дубликаты.Самый простой способ сделать это - добавить условное выражение в конец выражения генератора следующим образом:

myList= [1, 2, 3, 4, 5]
unorderedPairGenerator = ((x, y) for x in myList for y in myList if y > x)
for pair in unorderedPairGenerator:
    print(pair)
#(1, 2)
#(1, 3)
#(1, 4)
#(1, 5)
#(2, 3)
#(2, 4)
#(2, 5)
#(3, 4)
#(3, 5)
#(4, 5)

В качестве отдельного вопроса мне задали вопрос, аналогичный этому, в программном интервью, где они хотелимне сгенерировать все пары из списка чисел (без использования библиотечной функции).Сначала я сгенерировал все упорядоченные пары, а затем они попросили меня сгенерировать только неупорядоченные пары.Умение работать с перестановками и комбинациями - довольно важный навык кодирования!

Answer 7

Создание набора пар (четные, нечетные) комбинации

>>> a = { (i,j) for i in range(0,10,2) for j in range(1,10,2)}  
>>> a
{(4, 7), (6, 9), (0, 7), (2, 1), (8, 9), (0, 3), (2, 5), (8, 5), (4, 9), (6, 7), (2, 9), (8, 1), (6, 3), (4, 1), (4, 5), (0, 5), (2, 3), (8, 7), (6, 5), (0, 1), (2, 7), (8, 3), (6, 1), (4, 3), (0, 9)}

def combinations(lista, listb):
    return { (i,j) for i in lista for j in listb }

>>> combinations([1,3,5,6],[11,21,133,134,443])
{(1, 21), (5, 133), (5, 11), (5, 134), (6, 11), (6, 134), (1, 443), (3, 11), (6, 21), (3, 21), (1, 133), (1, 134), (5, 21), (3, 134), (5, 443), (6, 443), (1, 11), (3, 443), (6, 133), (3, 133)}