从较小的原始列表中返回 N 个随机项目



我很惊讶我没有找到一种没有循环的方法,因为这似乎是一个非常标准的问题。
在蟒蛇中工作,我有colors = ["red","green","blue"],我想将这些元素以随机顺序放入长度为 N 的列表中。 现在我正在使用:

import random
colors = ["red","green","blue"]
otherList = []
for i in range (10):  # N=10 
    otherList.append(random.choice(colors))

这返回:otherList = ["red","green","green","green","blue","green","red","green","green","blue"],这正是我想要的。我只是在寻找一种更惯用的方法?有什么想法吗?看起来 random.sample 可能是答案,但我在文档中没有看到任何完全符合我需求的内容。

您可以使用列表推导:

[random.choice(colors) for i in range(10)] #xrange for python2 compatability

或者random.sample()一些体操:

nrandom = 10
random.sample( colors*(nrandom//len(colors)+1), nrandom )  

虽然我不认为这比列表理解更好......

>>> import random
>>> colors = ["red","green","blue"]
>>> [random.choice(colors) for i in range(10)]
['green', 'green', 'blue', 'red', 'red', 'red', 'green', 'red', 'green', 'blue']

最新更新