Question

如何浏览list（数字）并获取其每一对可能的值？

示例（java）：

for(int i = 0; i < 5; i++)
    for(int j = i+1; j < 5; j++)
        print(i, j); // 01, 02, 03, 04, 12, 13, 14, 23, 24, 34

我的问题是列表是通过生成器函数获取的，因此很懒。此外，它可能非常大，因此将所有对保留在内存中不仅仅是绝望的解决方案。

所以最后的问题 - 如何在python中实现相同的行为，记住内存是有限的？此外，复杂性不应超过O(N^2)，其中N是list的长度。

Answer 1

实际上你想要这些组合，你可以使用itertools.combinations功能：

>>> from itertools import combinations
>>> list(combinations(range(5),2))
[(0, 1), (0, 2), (0, 3), (0, 4), (1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]

如果要连接结果，可以将元素转换为字符串，然后join将它们转换为：

>>> [''.join(map(str,i)) for i in combinations(range(5),2)]
['01', '02', '03', '04', '12', '13', '14', '23', '24', '34']

另外，作为获得预期输出的更有效方法，您可以使用包含0到5之间所有数字的字符串：

>>> from string import digits
>>> [''.join(i) for i in combinations(digits[:5],2)]
['01', '02', '03', '04', '12', '13', '14', '23', '24', '34']

Answer 2

您可以使用itertools.combinations：

执行此操作

list(itertools.combinations(range(5), 2))
Out[7]: 
[(0, 1),
 (0, 2),
 (0, 3),
 (0, 4),
 (1, 2),
 (1, 3),
 (1, 4),
 (2, 3),
 (2, 4),
 (3, 4)]

或者从对中制作字符串：

[''.join(str(i) for i in p) for p in itertools.combinations(range(5), 2)]
Out[8]: ['01', '02', '03', '04', '12', '13', '14', '23', '24', '34']

Answer 3

像这样（没有库）：

lst = [0, 1, 2, 3, 4]

newlst = [(y,z) for y in lst for z in lst]
output = set(['%s%s' % (x[0], x[1]) for x in newlst])

或者，作为一个单行（即不可读）：

set(['%s%s' % (x[0], x[1]) for x in [(y, z) for y in lst for z in lst]])

Answer 4

我认为我们都忽视了最简单的解决方案......

from itertools import combinations
for x in combinations(range(5), 2):
    print x

Answer 5

试试这个

from itertools import combination
>>>[''.join(map(str, i)) for i in combinations(range(5), 2)]
['01', '02', '03', '04', '12', '13', '14', '23', '24', '34']

如何遍历列表，获取每个可能的值对？

5 个答案: