python - 对列表进行重复采样

Question

我有一个列表 range(n) 并希望在不替换的情况下随机抽取 r 样本，即 np.random.choice(n,r ，替换= False)。但我想经常这样做，有没有一个快速的方法来执行以下命令:

a = [np.random.choice(n,r,replace = False) for i in range(100)]

注意:n 和 r 可以是“large”，s.t. a = np.random.choice(n,(r,100), Replace = True) 并删除具有两个相似索引的那些将是低效的。

Answer 1

我经常使用的一个技巧来替换 np.random.choice(..., replace=False) 的迭代正在生成2D随机数数组，然后使用 argsort/argpartition获取唯一的整数。

因此，与 argsort ，这将是 -

np.random.rand(100,n).argsort(axis=1)[:,:r]

或者，使用 r比 n 相对较小的数字，我们会使用argpartition为了提高性能，就像这样 -

np.argpartition(np.random.rand(100,n),r, axis=1)[:,:r]