使用 itertools.combinations 生成对时出现 Python 内存错误

Question

我有一个未指定数字的 int 列表。我想找到列表中匹配特定值的两个整数之间的差异。

from itertools import combinations

#Example of a list
intList = [3, 6, 2, 7, 1]
diffList = [abs(a -b) for a, b in combinations(intList, 2)]

#Given if difference = 2    
print diffList.count(2)

代码片段有效，但是当给出更大的列表时，我收到 MemoryError。谁能告诉我代码是否有问题或错误是由于我的硬件限制造成的？

Answer 1

“更大的列表”到底有多大？如果len(intList)为n，则len(diffList)将为n*(n-1)//2(一次取 2 个 n 事物的组合数)。如果 n 足够大，这将占用您的所有内存。

如果你只关心值2，

print sum(abs(a-b) == 2 for a, b in combinations(intList, 2))

是一种无论 intList 有多大都消耗很少内存的方法。但是，它仍然需要 时间 与 len(intList) 的平方成正比。