Python，来自向量的非零索引的随机样本

Question

我有一个叫做访问的一维矢量。我想找到其零位置的随机样本k指数。

这就是我现在所做的：

random.sample(np.where( visited == 0)[0]) , k) 

有更好的方法吗？

由于

Answer 1

你这样做的方式是合理的。但是，您可以使用更直观的nonzero函数：

random.sample(visited.nonzero(), k)

编辑：

关于评论中的第二个问题，您可以反转数组的“zeroness”：visited==0。你得到：

random.sample((visited==0).nonzero(), k)