Как случайным образом выбрать элемент из списка?


1761

Предположим, у меня есть следующий список:

foo = ['a', 'b', 'c', 'd', 'e']

Какой самый простой способ получить элемент в случайном порядке из этого списка?

Ответы:


2680

использование random.choice()

import random

foo = ['a', 'b', 'c', 'd', 'e']
print(random.choice(foo))

Для криптографически безопасного случайного выбора (например, для генерации ключевой фразы из списка слов) используйтеsecrets.choice()

import secrets

foo = ['battery', 'correct', 'horse', 'staple']
print(secrets.choice(foo))

secretsвпервые в Python 3.6, в более старых версиях Python вы можете использовать random.SystemRandomкласс:

import random

secure_random = random.SystemRandom()
print(secure_random.choice(foo))

3
Делает ли два последовательных вызова random.choice(foo)возврата два разных результата?
Эдуардо Пиньятелли

34
@EduardoPignatelli Каждый выбор является случайным, поэтому он может возвращать два разных результата, но в зависимости от начального начального числа это не гарантируется. Если вы хотите выбрать n отдельных случайных элементов из списка lst , используйтеrandom.sample(lst, n)
Graham

6
на связанной ноте, Standard pseudo-random generators are not suitable for security/cryptographic purposes. ссылка
Сяо

184

Если вы хотите случайным образом выбрать более одного элемента из списка или выбрать элемент из набора, я бы рекомендовал использовать random.sampleвместо этого.

import random
group_of_items = {1, 2, 3, 4}               # a sequence or set will work here.
num_to_select = 2                           # set the number to select here.
list_of_random_items = random.sample(group_of_items, num_to_select)
first_random_item = list_of_random_items[0]
second_random_item = list_of_random_items[1] 

Однако, если вы извлекаете только один элемент из списка, выбор будет менее неуклюжим, поскольку использование примера будет иметь синтаксис random.sample(some_list, 1)[0]вместо random.choice(some_list).

К сожалению, выбор работает только для одного вывода из последовательностей (таких как списки или кортежи). Хотя random.choice(tuple(some_set))может быть вариант для получения одного элемента из набора.

РЕДАКТИРОВАТЬ: Использование секретов

Как уже отмечали многие, если вам нужны более безопасные псевдослучайные образцы, вы должны использовать модуль секретов:

import secrets                              # imports secure module.
secure_random = secrets.SystemRandom()      # creates a secure random object.
group_of_items = {1, 2, 3, 4}               # a sequence or set will work here.
num_to_select = 2                           # set the number to select here.
list_of_random_items = secure_random.sample(group_of_items, num_to_select)
first_random_item = list_of_random_items[0]
second_random_item = list_of_random_items[1]

РЕДАКТИРОВАТЬ: Pythonic One-Liner

Если вы хотите более питонную однострочность для выбора нескольких элементов, вы можете использовать распаковку.

import random
first_random_item, second_random_item = random.sample(group_of_items, 2)

1
BTW secretsмодуль был добавлен в стандартную библиотеку Python в версии 3.6 python.org/dev/peps/pep-0506
1

160

Если вам также нужен индекс, используйте random.randrange

from random import randrange
random_index = randrange(len(foo))
print(foo[random_index])

42

Начиная с Python 3.6 вы можете использовать secretsмодуль, который предпочтительнее randomмодуля для криптографии или безопасности.

Чтобы распечатать случайный элемент из списка:

import secrets
foo = ['a', 'b', 'c', 'd', 'e']
print(secrets.choice(foo))

Чтобы напечатать случайный индекс:

print(secrets.randbelow(len(foo)))

Подробнее см. PEP 506 .


33

Я предлагаю скрипт для удаления случайно выбранных элементов из списка, пока он не станет пустым:

Сохраните setи удалите случайно выбранный элемент (с choice), пока список не станет пустым.

s=set(range(1,6))
import random

while len(s)>0:
  s.remove(random.choice(list(s)))
  print(s)

Три прогона дают три разных ответа:

>>> 
set([1, 3, 4, 5])
set([3, 4, 5])
set([3, 4])
set([4])
set([])
>>> 
set([1, 2, 3, 5])
set([2, 3, 5])
set([2, 3])
set([2])
set([])

>>> 
set([1, 2, 3, 5])
set([1, 2, 3])
set([1, 2])
set([1])
set([])

20
Или вы могли бы просто раз и либо итерация его или вытолкнуть его для получения результатов. Любой из них приведет к совершенно адекватному потоку «случайного выбора без повторов», просто случайность будет введена в начале. random.shufflelist
ShadowRanger

2
Теоретически вы можете использовать метод pop () набора, чтобы удалить произвольный элемент из набора и вернуть его, но он, вероятно, недостаточно случайный.
Джубарк

14
foo = ['a', 'b', 'c', 'd', 'e']
number_of_samples = 1

В питоне 2:

random_items = random.sample(population=foo, k=number_of_samples)

В питоне 3:

random_items = random.choices(population=foo, k=number_of_samples)

6
Обратите внимание, что random.choicesс заменой пока random.sampleбез замены.
CentAu

1
Также обратите внимание, что random.choices доступен с версии 3.6 и выше, но не раньше!
Кирилл Н.

11

numpy решение: numpy.random.choice

Для этого вопроса он работает так же, как принятый ответ ( import random; random.choice()), но я добавил его, потому что программист, возможно, уже импортировал numpy(как я), а также есть некоторые различия между этими двумя методами, которые могут касаться вашего фактического варианта использования.

import numpy as np    
np.random.choice(foo) # randomly selects a single item

Для воспроизводимости вы можете сделать:

np.random.seed(123)
np.random.choice(foo) # first call will always return 'c'

Для образцов одного или нескольких элементов , возвращаемых как array, передайте sizeаргумент:

np.random.choice(foo, 5)          # sample with replacement (default)
np.random.choice(foo, 5, False)   # sample without replacement

9

Как случайным образом выбрать элемент из списка?

Предположим, у меня есть следующий список:

foo = ['a', 'b', 'c', 'd', 'e']  

Какой самый простой способ получить элемент в случайном порядке из этого списка?

Если вы хотите, чтобы значение было действительно случайным , то я предлагаю secrets.choiceиз стандартной библиотеки (Новое в Python 3.6.):

>>> from secrets import choice         # Python 3 only
>>> choice(list('abcde'))
'c'

Вышесказанное эквивалентно моей предыдущей рекомендации, использующей SystemRandomобъект из randomмодуля с choiceметодом, доступным ранее в Python 2:

>>> import random                      # Python 2 compatible
>>> sr = random.SystemRandom()
>>> foo = list('abcde')
>>> foo
['a', 'b', 'c', 'd', 'e']

И сейчас:

>>> sr.choice(foo)
'd'
>>> sr.choice(foo)
'e'
>>> sr.choice(foo)
'a'
>>> sr.choice(foo)
'b'
>>> sr.choice(foo)
'a'
>>> sr.choice(foo)
'c'
>>> sr.choice(foo)
'c'

Если вы хотите детерминированный псевдослучайный выбор, используйте choiceфункцию (которая на самом деле является связанным методом для Randomобъекта):

>>> random.choice
<bound method Random.choice of <random.Random object at 0x800c1034>>

Это кажется случайным, но на самом деле это не так, что мы можем увидеть, если повторить его повторно:

>>> random.seed(42); random.choice(foo), random.choice(foo), random.choice(foo)
('d', 'a', 'b')
>>> random.seed(42); random.choice(foo), random.choice(foo), random.choice(foo)
('d', 'a', 'b')
>>> random.seed(42); random.choice(foo), random.choice(foo), random.choice(foo)
('d', 'a', 'b')
>>> random.seed(42); random.choice(foo), random.choice(foo), random.choice(foo)
('d', 'a', 'b')
>>> random.seed(42); random.choice(foo), random.choice(foo), random.choice(foo)
('d', 'a', 'b')

Комментарий:

Дело не в том, является ли random.choice действительно случайным или нет. Если вы почините семя, вы получите воспроизводимые результаты - и для этого семя предназначено. Вы также можете передать семена в SystemRandom.sr = random.SystemRandom(42)

Да, вы можете передать ему аргумент "seed", но вы увидите, что SystemRandomобъект просто игнорирует его :

def seed(self, *args, **kwds):
    "Stub method.  Not used for a system random number generator."
    return None

8

если вам нужен индекс, просто используйте:

import random
foo = ['a', 'b', 'c', 'd', 'e']
print int(random.random() * len(foo))
print foo[int(random.random() * len(foo))]

random.choice делает то же самое :)


2
@tc. На самом деле, он делает то же самое. Реализация random.choice(self, seq)есть return seq[int(self.random() * len(seq))].
Вим

2
@wim Это немного разочаровывает, но очень разочаровывает то, что это также определение того, randrange()что означает, например, random.SystemRandom().randrange(3<<51)демонстрирует значительный уклон. Вздох ...
тс.

6
@ kevinsa5 В конечном счете это потому, что a float(двойное значение IEEE) может принимать только конечное число значений в [0,1). Random.random()генерирует свой вывод традиционным способом: выберите случайное целое число [0, 2**53)и разделите на 2**53(53 - число бит в двойном). Таким образом, random()возвращает 2 ** 53 равновероятных двойных чисел, и вы можете разделить это равномерно на N выходов, только если N - степень 2. Смещение мало для малого N, но посмотрите collections.Counter(random.SystemRandom().randrange(3<<51)%6 for i in range(100000)).most_common(). (Java. Random.nextInt () избегает такого смещения.)
tc.

1
@tc. Я полагаю, что что-то меньшее, чем около 2**40(то есть 1099511627776), будет достаточно маленьким, чтобы смещение не имело значения на практике? Это действительно должно быть указано в документации, потому что если кто-то не будет дотошным, он может не ожидать проблем с этой частью своего кода.
Евгений Сергеев

@tc .: На самом деле, randomиспользуется getrandbitsдля получения достаточного количества бит для генерации результата для больших randranges ( random.choiceтакже использует это). Это верно как для 2.7, так и для 3.5. Используется только self.random() * len(seq)когда getrandbitsнедоступно. Это не делает глупость, о которой ты думаешь.
ShadowRanger

7

Это код с переменной, которая определяет случайный индекс:

import random

foo = ['a', 'b', 'c', 'd', 'e']
randomindex = random.randint(0,len(foo)-1) 
print (foo[randomindex])
## print (randomindex)

Это код без переменной:

import random

foo = ['a', 'b', 'c', 'd', 'e']
print (foo[random.randint(0,len(foo)-1)])

И это код в кратчайшем и умном способ сделать это:

import random

foo = ['a', 'b', 'c', 'd', 'e']
print(random.choice(foo))

(Python 2.7)


3

Следующий код демонстрирует, нужно ли вам производить одинаковые предметы. Вы также можете указать, сколько образцов вы хотите извлечь. Метод возвращает новый список , содержащий элементы населения, оставляя исходное население без изменений. Результирующий список находится в порядке выбора, так что все подрезы также будут действительными случайными выборками.
sample

import random as random
random.seed(0)  # don't use seed function, if you want different results in each run
print(random.sample(foo,3))  # 3 is the number of sample you want to retrieve

Output:['d', 'e', 'a']

1

Случайный выбор товара:

import random

my_list = [1, 2, 3, 4, 5]
num_selections = 2

new_list = random.sample(my_list, num_selections)

Чтобы сохранить порядок списка, вы можете сделать:

randIndex = random.sample(range(len(my_list)), n_selections)
randIndex.sort()
new_list = [my_list[i] for i in randIndex]

Дубликат https://stackoverflow.com/a/49682832/4383027


0

Мы также можем сделать это, используя randint.

from random import randint
l= ['a','b','c']

def get_rand_element(l):
    if l:
        return l[randint(0,len(l)-1)]
    else:
        return None

get_rand_element(l)

19
С какой стати ты так поступил, когда есть random.choice()и random.randrange()?
Алексис

«random.choice ()» выдаст вам «IndexError: список индексов вне диапазона» в пустом списке.
Абдул Маджид

6
Как и должно быть: для этого есть исключения. Выбор из пустого списка является ошибкой. Возврат Noneтолько пинает банку в какую-то случайную точку позже, когда недопустимый «элемент» вызывает исключение; или, что еще хуже, вместо исключения вы получаете неверную программу и даже не знаете об этом.
Alexis

Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.