Python:从数组中选择与特定值的元素相对应的索引

Question

我有一个如下所示的数组:

x = [1, 1, 2, 3, 3, 2, 2, 1, 2, 3, 2, 3, 2, 1, 2, 1, 1, 2, 1]

我想编写一个函数，它将随机返回与指定数字相对应的一些指定数量的索引。换句话说，如果我向函数传递数组 x、所需的索引数量(例如 3)和目标值 1，我希望它返回一个数组，例如:

[0,7,13]

由于 0、7 和 13 是与 x 中的 1 对应的索引.

有谁知道我如何有效地做到这一点？

Answer 1

您想为此使用random.sample:

import random

def f(arr, target, num): 
    return random.sample([i for i, x in enumerate(arr) if x == target], k=num) 

x = [1, 1, 2, 3, 3, 2, 2, 1, 2, 3, 2, 3, 2, 1, 2, 1, 1, 2, 1]                                   
print(f(x, 1, 3))                                                                                   

输出:

[0, 1, 15]