生成对的pythonic方法

我想要下面的代码，但要“pythonic”样式或使用标准库：

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

在组合学的意义上，这些并不是真正的“组合”，而是来自

a

和

b

的笛卡尔积的元素。标准库中生成这些对的函数是：

该库具有组合功能。如Sven所述，

itertools.product

将是这种情况下的适当功能：

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

---

嵌套生成器表达式也可以工作：

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

---

正如@Sven所说，您的代码试图获取列表

a

和

b

中所有有序的元素对。在本例中，

itertools。产品（a，b）

就是您想要的。如果您实际上想要的是“组合”，它们都是列表

a

中不同元素的无序对，那么您想要的是

itertools.compositions（a，2）

创建一组成对（偶数、奇数）组合

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

---

我们可能会问一个问题，您是要生成所有有序对还是所有无序对。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)

---

另一方面，在一次软件采访中，我被问到了一个类似的问题，他们希望我从一个数字列表中生成所有对（不使用库函数）。首先我生成了所有有序对，然后他们让我只生成无序对。能够处理排列和组合是一项非常重要的编码技能

你能提供一些输入和输出示例吗？当前，您正在为

a

和

b

中的每个元素创建一对。这真的是你想要的吗？不是标准的库函数，而是比我的类似C的版本更具python风格；-）感谢您提供itertools.product（）的替代方案@Dima：至少您使用了

yield

；）我还想要（1,5），（1,6），（2,4），（2,6），（3,4），（3,5）以及zip创建的三对……虽然@fransua获得了-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)

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

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)