Python：如何将生成器迭代值附加到列表中

Question

我有一个简单的生成器，可以给我一组坐标的排列。 我希望使用以下代码将每个新排列保存到数组中的一个元素：

import random
def poss_comb(coord):
    spin=random.shuffle
    if spin:
        spin(coord)
        yield (coord)
...

a=[]
for n in xrange(0,10):
    for item in poss_comb(coord):
        print item
        a.append(item)

但是，当打印结果时，打印 item 给了我我想要的：

['0 1', '', '1 2', '1 3']
['0 1', '', '1 2', '1 3']
['1 2', '0 1', '1 3', '']
['0 1', '1 2', '', '1 3']
['1 3', '', '1 2', '0 1']
['1 3', '1 2', '0 1', '']
['0 1', '', '1 3', '1 2']
['1 2', '0 1', '', '1 3']
['1 2', '1 3', '', '0 1']
['', '1 2', '1 3', '0 1']

而打印list a 提供了一个数组，其中每个元素都是最后一个排列的副本。

有什么更好的方法来做到这一点？

Answer 1

您的生成器不会产生 new 列表，它会一遍又一遍地产生 相同的列表。当您将产生的引用附加到 a 时，您只会一遍又一遍地看到相同的原始列表，它是最近改组的形式。

改为生成副本：

def poss_comb(coord):
    coord = coord[:]  # use a local copy of the list
    random.shuffle(coord)
    yield coord

或者创建一个随机排序，而不是使用 sorted() 函数的就地改组：

def poss_comb(coord):
    yield sorted(coord, key=lambda k: random.random())