找到组合的更快方法？

Question

我试图找到np.linspace(0,n,n*10+1) 子区间大于宽度的所有可能的子区间（比如宽度 = 0.5）

所以我通过使用 itertools 尝试了这个

import itertools
ranges=np.linspace(0,n,n*10+1)
#find all combinations
combinations=list(itertools.combinations(ranges,2))
#using for-loops to calculate width of each intervals
#and append to new list if the width is greater than 0.5
save=[]
for i in range(len(combinations)):
    if combinations[i][1]-combinations[i][0]>0.5:
        save.append(combinations[i]) 

但这需要太多次，尤其是当 n 变大时，尤其会消耗大量内存

所以我想知道我是否可以在收集组合时更快地修改函数或设置约束

Answer 1

itertools.combinations(...)返回一个生成器，这意味着返回的对象在需要时生成其值，而不是一次计算所有内容并将结果存储在内存中。您可以通过将其转换为列表来强制立即计算和存储，但这是不必要的。简单地迭代combinations对象而不是制作它的列表并迭代索引（无论如何都不应该这样做）：

import itertools
ranges=np.linspace(0,n,n*10+1)  # alternatively 'range(100)' or so to test
combinations=itertools.combinations(ranges,2)
save=[]

for c in combinations:
    if c[1] - c[0] > 0.5:
        save.append(c)