从列表中进行互斥随机抽样

input = ['beleriand','mordor','hithlum','eol','morgoth','melian','thingol']

我在创建 X 个大小为 Y 的列表而不重复任何元素时遇到问题。

我一直在做的是使用：

x = 3
y = 2

import random

output = random.sample(input, y)
# ['mordor', 'thingol']

但如果我重复这个，那么我就会重复。

我希望输出是这样的

[['mordor', 'thingol'], ['melian', 'hithlum'], ['beleriand', 'eol']]

自从我选择了x = 3(3 列表) 尺寸y = 2（每个列表 2 个元素）。

def random_generator(x,y):
    ....

您可以简单地打乱原始列表，然后生成n组的m元素依次来自它。可能存在比该数量更少或更多的组。注意input是Python内置函数的名称，所以我重命名了它words.

import itertools
from pprint import pprint
import random

def random_generator(seq, n, m):
    rand_seq = seq[:]  # make a copy to avoid changing input argument
    random.shuffle(rand_seq)
    lists = []
    limit = n-1
    for i,group in enumerate(itertools.izip(*([iter(rand_seq)]*m))):
        lists.append(group)
        if i == limit: break  # have enough
    return lists

words = ['beleriand', 'mordor', 'hithlum', 'eol', 'morgoth', 'melian', 'thingol']
pprint(random_generator(words, 3, 2))

Output:

[('mordor', 'hithlum'), ('thingol', 'melian'), ('morgoth', 'beleriand')]

迭代地生成组会更Pythonic。上面的函数可以很容易地变成生成器yield每个组都一一返回，而不是将它们全部返回到相对较长的列表中：

def random_generator_iterator(seq, n, m):
    rand_seq = seq[:]
    random.shuffle(rand_seq)
    limit = n-1
    for i,group in enumerate(itertools.izip(*([iter(rand_seq)]*m))):
        yield group
        if i == limit: break

pprint([group for group in random_generator_iterator(words, 3, 2)])