作为练习,我一直在尝试各种方法来生成Python中列表的所有排列 - 递归,非递归...... - 并将性能与itertools.permutations()
进行比较。但我在使用递归方法的生成器版本时遇到问题,该方法没有StopIteration
异常干净地完成,而是抛出IndexError
:
def spawnperms(alist):
"""same algorithm as recursive option, but a generator"""
if (alist == []):
yield []
for perm in spawnperms(alist[:-1]):
for i in range(len(perm)+1):
yield perm[:i] + [alist[-1]] + perm[i:]
从Python解释器调用它:
>>> for i in spawnperms(range(3)):
... print i
...
[2, 1, 0]
[1, 2, 0]
[1, 0, 2]
[2, 0, 1]
[0, 2, 1]
[0, 1, 2]
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 5, in spawnperms
File "<stdin>", line 5, in spawnperms
File "<stdin>", line 5, in spawnperms
File "<stdin>", line 7, in spawnperms
IndexError: list index out of range
哎哟。我尝试用pdb
逐步完成它,这几乎在我的大脑中创建了一个堆栈溢出,但我所理解的是递归“向下”到空列表,然后是外部(我认为){{1循环用完了索引。
如何更正我的代码?
编辑:从Mark Byers那里学到的一个看似简单的正确答案是 干净的编码实践可以防止错误 。如果我在for
之后系统地使用else
,无论我是否认为可以重新审视这种情况,这都不会发生。它仍然感觉非常愚蠢!
答案 0 :(得分:6)
您错过了else
:
if (alist == []):
yield []
else:
for ...
这是因为yield
的行为方式与return
不同。当您请求下一个值时,在yield
语句后继续执行。