web-dev-qa-db-ja.com

ペアを生成するPythonの方法

以下のコードのようなものが欲しいですが、「Python」スタイルまたは標準ライブラリを使用します：

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

これらは実際には組み合わせ論的な意味での「組み合わせ」ではなく、aとbのデカルト積の要素です。これらのペアを生成する標準ライブラリの関数は、 itertools.product() ：

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

@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)
27
Rob

itertools ライブラリには組み合わせ関数があります。スヴェンが述べたように、itertools.productは、この場合の適切な関数です。

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

ネストされたジェネレーター式も機能します。

product = ((i, j) for i in a for j in b)
for i, j in product:
    # ...
7
mhyfritz
>>>a=[1,2,3]
>>>b=[4,5,6]
>>>Zip(a,b)
[(1, 4), (2, 5), (3, 6)]
4
fransua

ペア（偶数、奇数）の組み合わせのセットを作成します

>>> 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)}
1
wwright

質問するかもしれないのは、すべての順序付きペアを生成するか、すべての順序なしペアを生成するかです。 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)

余談ですが、ソフトウェアインタビューでこれに似た質問が出されました。そこでは、（ライブラリ関数を使用せずに）数値のリストからすべてのペアを生成するように求められました。最初にすべての順序付きペアを生成し、次に、順序付けられていないペアのみを生成するように要求しました。順列と組み合わせで作業できることは、非常に重要なコーディングスキルです！

1
lonknex