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']

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

最好的办法是什么？

Answer 1

您的生成器不会产生新的列表，它会一遍又一遍地生成相同的列表。当你追加那个产生a的引用时，你只能一遍又一遍地看到相同的原始列表，就像它最近洗牌一样。

改为生成副本：

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

或创建随机排序，而不是使用sorted()函数进行inplace shuffling：

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

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

1 个答案: