找到组合的更快方法？

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 变大时，尤其是它会消耗大量 ram

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

Answer 1

itertools.combinations(...) returns 一个生成器，这意味着返回的对象在需要时产生它的值，而不是一次计算所有内容并将结果存储在内存中。您通过将其转换为列表来强制立即计算和存储，但这是不必要的。简单地遍历 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)

找到组合的更快方法？

faster way of finding combinations?

python

combinations

itertools

python-3.x