Блог

Я очень хорошо знаю, как выбрать случайный элемент из списка с помощью random.choice(seq) , но откуда мне узнать индекс этого элемента?

 import random
l = ['a','b','c','d','e']
i = random.choice(range(len(l)))
print i, l[i]
  

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

 >>> import random
>>> a = [1, 2, 3, 4, 5]
>>> index = random.randint(0,len(a)-1)
>>> index
0
>>> a[index]
1
  

Вы можете сделать это с помощью функции randrange из модуля random

 import random
l = ['a','b','c','d','e']
i = random.randrange(len(l))
print i, l[i]
  

Самый элегантный способ сделать это — random.randrange:

 index = random.randrange(len(MY_LIST))
value = MY_LIST[index]
  

Это также можно сделать в python3, менее элегантно (но все же лучше, чем .index ) с помощью random.choice для объекта range:

 index = random.choice(range(len(MY_LIST)))
value = MY_LIST[index]
  

Единственными допустимыми решениями являются это решение и random.randint решения.

Те, которые используют, list.index не только работают медленно ( O(N) для каждого поиска, а не O(1) ; становится действительно плохо, если вы делаете это для каждого элемента, вам придется делать O(N^2) сравнения), но также вы получите искаженные / неправильные результаты, если элементы списка не уникальны.

Можно было бы подумать, что это медленно, но оказывается, что это лишь немного медленнее, чем другое правильное решение random.randint , и может быть более читаемым. Я лично считаю это более элегантным, потому что не нужно манипулировать числовым индексом и использовать ненужные параметры, как это имеет отношение к randint(0,len(...)-1) , но некоторые могут счесть это особенностью, хотя нужно знать randint соглашение о включающем диапазоне [start, stop] .

Доказательство скорости для random.choice: единственная причина, по которой это работает, заключается в том, что range объект ОПТИМИЗИРОВАН для индексации. В качестве доказательства вы можете сделать random.choice(range(10**12)) ; если бы он повторялся по всему списку, ваша машина была бы замедлена до обхода.

редактировать: я пропустил randrange, потому что в документах, казалось, говорилось «не используйте эту функцию» (но на самом деле это означало «эта функция pythonic, используйте ее»). Спасибо martineau за указание на это.

Вы могли бы, конечно, абстрагировать это в функцию:

 def randomElement(sequence):
    index = random.randrange(len(sequence))
    return index,sequence[index]

i,value = randomElement(range(10**15))  # try THAT with .index, heh
                                        # (don't, your machine will die)
                                        # use xrange if using python2
# i,value = (268840440712786, 268840440712786)
  

Если значения уникальны в последовательности, вы всегда можете сказать: list.index(value)

Использование randrage(), как было предложено, является отличным способом получить индекс. Создав словарь, созданный с помощью понимания, вы можете сократить этот код до одной строки, как показано ниже. Обратите внимание, что, поскольку в этом словаре есть только один элемент, при вызове popitem() вы получаете комбинированный индекс и значение в виде кортежа.

 import random

letters = "abcdefghijklmnopqrstuvwxyz"

# dictionary created via comprehension
idx, val = {i: letters[i] for i in [random.randrange(len(letters))]}.popitem()

print("index {} value {}" .format(idx, val))
  

Мы также можем использовать метод sample(). Если вы хотите случайным образом выбрать n элементов из списка

 import random
l, n = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 2
index_list = random.sample(range(len(l)), n)
  

index_list будет иметь уникальные индексы.

Я предпочитаю sample() вместо choices(), поскольку sample () не допускает дублирования элементов в последовательности.