生成& # 39;n # 39;范围内唯一的随机数

我知道如何在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出现了两次。

543177 次浏览
小开

首先生成数据的范围，然后像这样打乱它

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

通过这样做，您将得到特定范围内的所有数字，但顺序是随机的。

但是你可以使用random.sample来获取你需要的元素的数量，从一个像这样的数字范围

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

你可以添加到set，直到你到达n:

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

注意要有一个比n更小的范围。它将永远循环，无法找到新的数字插入到n

小开
最佳答案

如果您只需要采样而不需要更换:

>>> 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
小开

你可以使用标准库中的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)

提交答案