我如何遍历列表(数字)并获取每对可能的值?
例子(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 的长度。
请您参考如下方法:
其实你想要组合,你可以使用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(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']
