python 如何获得成对的值,其中第一个值取自一个列表,第二个值取自另一个列表?[duplicate]

iih3973s  于 2023-02-18  发布在  Python
关注(0)|答案(7)|浏览(123)
    • 此问题在此处已有答案**:

(17个答案)
5小时前关门了。
我想像下面的代码,但"Python"风格或使用标准库:

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

o8x7eapl1#

从组合学的Angular 来看,这些并不是真正的“组合”,而是ab的笛卡尔积中的元素。标准库中生成这些对的函数是itertools.product()

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

3yhwsihp2#

正如@Sven所说,您的代码试图获取列表ab的所有有序元素对,在这种情况下,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)
ct3nt3jp

ct3nt3jp3#

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

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

bvhaajcl4#

itertools库中有一些组合函数,如Sven所说,itertools.product是适合这种情况的函数:

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

laximzn55#

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

cgvd09ve6#

我们可能会问的一个问题是,你想生成所有有序对还是所有无序对,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)
hfyxw5xn

hfyxw5xn7#

创建成对(偶数、奇数)组合集

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

相关问题