python - 如何获取 numpy.random.choice 的索引？ - Python

Question

是否可以修改 numpy.random.choice 函数以使其返回所选元素的索引？基本上，我想创建一个列表并随机选择元素而不替换

import numpy as np
>>> a = [1,4,1,3,3,2,1,4]
>>> np.random.choice(a)
>>> 4
>>> a
>>> [1,4,1,3,3,2,1,4]

a.remove(np.random.choice(a)) 将删除列表中具有它遇到的值的第一个元素(a[1] 在上面的例子)，它可能不是被选择的元素(例如，a[7])。

Answer 1

这是找出随机选择元素的索引的一种方法:

import random # plain random module, not numpy's
random.choice(list(enumerate(a)))[0]
=> 4      # just an example, index is 4

或者您可以在一个步骤中检索元素和索引:

random.choice(list(enumerate(a)))
=> (1, 4) # just an example, index is 1 and element is 4