生成一定范围内的‘n’个唯一随机数[重复]

问题描述：

我知道如何在 Python 中生成一个范围内的随机数。

random.randint(numLow, numHigh)

我知道我可以把它放在一个循环中来生成 n 个这样的数字

for x in range (0, n):
    listOfNumbers.append(random.randint(numLow, numHigh))

但是，我需要确保列表中的每个数字都是唯一的。除了大量条件语句之外，有没有一种直接的方法可以生成 n 个唯一的随机数？

重要的是，列表中的每个数字都与其他数字不同。

所以

[12, 5, 6, 1] = 好

但

[12, 5, 5, 1] = 不好，因为数字 5 出现了两次。

解决方案 1：

如果只是需要不放回抽样：

>>> import random
>>> random.sample(range(1, 100), 3)
[77, 52, 45]

random.sample取总体和样本大小k并返回k总体的随机成员。

k如果你必须控制大于的情况len(population)，那么你需要准备好捕捉ValueError：

>>> try:
...   random.sample(range(1, 2), 3)
... except ValueError:
...   print('Sample size exceeded population size.')
... 
Sample size exceeded population size

解决方案 2：

首先生成数据范围，然后像这样对其进行打乱

import random
data = list(range(numLow, numHigh))
random.shuffle(data)
print data

通过这种方式，您将获得特定范围内的所有数字，但顺序是随机的。

但是你可以使用它random.sample来从这样的数字范围内获取所需的元素数量

print random.sample(range(numLow, numHigh), 3)

解决方案 3：

您可以添加set直到达到n：

setOfNumbers = set()
while len(setOfNumbers) < n:
    setOfNumbers.add(random.randint(numLow, numHigh))

注意不要使用小于 的范围n。它将永远循环，无法找到要插入的新数字n

解决方案 4：

您可以使用标准库random.sample中的函数从总体中选择k 个元素：

import random
random.sample(range(low, high), n)

如果可能的数字范围相当大，则可以使用itertools.islice无限随机生成器：

import itertools
import random

def random_gen(low, high):
    while True:
        yield random.randrange(low, high)

gen = random_gen(1, 100)
items = list(itertools.islice(gen, 10))  # Take first 10 random elements

问题更新后，现在很清楚您需要n 个不同的（唯一）数字。

import itertools
import random

def random_gen(low, high):
    while True:
        yield random.randrange(low, high)

gen = random_gen(1, 100)

items = set()

# Try to add elem to set until set length is less than 10
for x in itertools.takewhile(lambda x: len(items) < 10, gen):
    items.add(x)