2011-06-27 19 views
16

çiftleri, ancak üretmek için pythonic yolu "pythonic" tarzı ya da standart kitaplığı kullanarak:aşağıda kodu gibi bir şey istiyorum

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

bazı örnek giriş ve çıkış verebilir misiniz (çift, tek) kombinasyon kümesi oluşturma? Şu anda 'a' ve 'b' öğelerinde her öğe için bir çift oluşturuyorsunuz. Bu gerçekten istediğin şey mi? –

cevap

26

Bu kombinatorik anlamında "kombinasyonlar" gerçekten değil, bunlar oldukça unsurlardır a ve b'un kartezyen ürününden. Bu çiftleri oluşturmak için standart kütüphane işlevi itertools.product() olup:

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

itertools kütüphane bir kombinasyon fonksiyonları vardır. Bir iç içe jeneratör ifadesinde çok çalışacak

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

: @Sven söylediği gibi

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

Standart kütüphane işlevi değil, daha çok pythonic benim için "C" -like sürümü ;-) itertools.product() – Dima

+1

@ Dima'ya alternatif sağladığınız için teşekkürler: En azından 'verim' kullandınız;) – mhyfritz

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

Ben de istiyorum (1 , 5), (1,6), (2,4), (2,6), (3,4), (3,5), zip tarafından yaratılan üç çifte ek olarak .... – Dima

+1

fransua -2 downvote aldı, bu tam olarak bir çift olarak ihtiyacım olan şey. Cevabını vereceğim. – shailenTJ

9

, kodunuz almak için çalışıyor Sven belirtti gibi, itertools.product bu durumda uygun fonksiyon olacaktır Listelenen tüm elemanların çiftleri a ve b. Bu durumda itertools.product(a,b) istediğin şeydir. Bunun yerine, aslında, "a listesinin farklı öğelerinin sırasız çiftleri olan" kombinasyonlar "ı istiyorsanız, o zaman itertools.combinations(a,2) olmasını istersiniz.

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

çiftleri

>>> 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)} 
İlgili konular