Python,2个列表之间所有可能的数字组合

时间:2013-09-28 22:14:32

标签: python list combinations

我是python的新手,我想要一些有问题的指导。

假设您有两个列表,长度相同且填充数字:

x=[0,1,2,3]
y=[0,4,5,6]

Python如何计算同一位置列表中数字之间的数字? 所以对于这个例子。

[0,1,2,3]
[0,2,2,3]
[0,3,2,3]
[0,4,2,3]
[0,1,3,3]
...

我希望有更多经验的人能对这件事有所了解。

1 个答案:

答案 0 :(得分:7)

x=[0,1,2]
y=[0,4,5]

from itertools import product

for lst in product(*(range(a, b+1) for a, b in zip(x, y))):
    print(lst)
#>>> (0, 1, 2)
#>>> (0, 1, 3)
#>>> (0, 1, 4)
#>>> (0, 1, 5)
#>>> (0, 2, 2)
#>>> (0, 2, 3)
#>>> (0, 2, 4)
#>>> (0, 2, 5)
#>>> (0, 3, 2)
#>>> (0, 3, 3)
#>>> (0, 3, 4)
#>>> (0, 3, 5)
#>>> (0, 4, 2)
#>>> (0, 4, 3)
#>>> (0, 4, 4)
#>>> (0, 4, 5)

zip获取xy并生成(在这种情况下)[(0, 0), (1, 4), (2, 5)]

for a, b in接受每一对并将其解压缩到ab,因此对于第二个a = 1b = 4

然后为每个人制作range。它是range(a, b+1),因为该范围中的最后一项已被删除。

然后你有一个可迭代的范围:

[ range(0, 1), range(1, 5), range(2, 6) ]

product需要一些迭代,并对所有可能的组合进行迭代。因此,您使用“*”运算符“解包”迭代项,因此product被称为

product(*[ range(0, 1), range(1, 5), range(2, 6) ])

相当于

product(range(0, 1), range(1, 5), range(2, 6))

并产生结果。