gpt4 book ai didi

python - 防止 itertools.permutation 中的内存错误

转载 作者:太空狗 更新时间:2023-10-29 22:02:03 25 4
gpt4 key购买 nike

首先我想提一下我有一个 3 GB 的内存。

我正在研究一种算法,该算法在节点上的时间呈指数级,因此我在代码中有它

perm = list( itertools.permutations(list(graph.Nodes))) # graph.Nodes is a tuple of 1 , 2 , ... n integers

它生成列表中的所有顶点组合,然后我可以处理其中一个排列。

但是,当我为 40 个顶点运行程序时,会出现内存错误。

有没有更简单的实现方式,通过它我可以生成顶点的所有组合并且没有这个错误。

最佳答案

尝试使用由排列生成的迭代器而不是用它重新创建一个列表:

perm_iterator = itertools.permutations(list(graph.Nodes))

for item in perm_iterator:
do_the_stuff(item)

通过这样做,python 将只在内存中保留当前使用的排列,而不是所有排列(就内存使用而言,它确实更好;))

另一方面,一旦内存问题得到解决,处理所有排列的时间将随着顶点的数量呈指数增长....

关于python - 防止 itertools.permutation 中的内存错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/6503388/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com