python生成器中的next()方法很奇怪

时间:2012-09-11 11:17:23

标签: python generator

def conflict(state, nextX):
    nextY = len(state)
    for i in range(nextY):
        if abs(state[i]-nextX) in (0, nextY-i):
            return True
    return False

def queens(num=8, state=()):
    for pos in range(num):
        if not conflict(state, pos):
            if len(state) == num-1:
                yield (pos,)
            else:
                results=queens(num, state + (pos,))
                for res in results:
                    yield (pos,) + res 

print list(queens(8))

这是八个皇后的解决方案,效果很好。但是当我做了一点改变时,对于python Generator中的next()方法变得奇怪了。

def conflict(state, nextX):
    nextY = len(state)
    for i in range(nextY):
        if abs(state[i]-nextX) in (0, nextY-i):
            return True
    return False

def queens(num=8, state=()):
    for pos in range(num):
        if not conflict(state, pos):
            if len(state) == num-1:
                yield (pos,)
            else:
                results=queens(num, state + (pos,))
                for res in results:
                    yield (pos,) +results.next()   //here is the change. res -->results.next()  

print list(queens(8))

只是打印一个[]。我多么尴尬!我无法理解为什么next()方法适用于任何其他情况但不在此处。

1 个答案:

答案 0 :(得分:3)

你期待发生什么?

当您在生成器上调用next时,它会使生成器前进并返回下一个项目。你的基本情况(len(state) == num-1)只产生一个项目,所以在这种情况下它将推进迭代器,接收StopIteration,并终止循环,所以你不会得到任何结果。

根本问题是你在迭代它时试图推进迭代器;这不太可能奏效。