随机翻转数组中的m值

Question

我有一个长度为n的数组，我想从中随机选择m个元素并翻转它们的值。最有效的方法是什么？在

有两种情况，m=1是一种特殊情况。可以单独讨论，并且m=/=1。在

我的尝试是：

import numpy as np
n = 20
m = 5
#generate an array a
a = np.random.randint(0,2,n)*2-1
#random choose `m` element and flip it.
for i in np.random.randint(0,n,m):
    a[m]=-a[m]

假设m是十，而{}是百。在

Answer 1

您可以对数组索引进行随机排列，取其中的第一个m并翻转它们的值：

a[np.random.permutation(range(len(a)))[:m]]*=-1

使用permutation验证您不会选择同一个索引两次。在