生成' n'一个范围内的唯一随机数

时间:2014-04-03 15:29:57

标签: python random unique

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

4 个答案:

答案 0 :(得分:289)

如果您只是需要取样而无需更换:

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

答案 1 :(得分:23)

首先生成数据范围,然后像这样随机播放

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

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

但您可以使用random.sample从这样的数字范围中获取所需的元素数量

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

答案 2 :(得分:12)

您可以添加到set,直至到达n

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

小心范围小于n的范围。它将永远循环,无法找到要插入n

的新数字

答案 3 :(得分:4)

您可以使用standard library中的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)