python - Numpy:替换数组中的随机元素

Question

我已经用谷歌搜索了一下，没有找到任何好的 答案。

问题是，我有一个 2d numpy 数组，我想 在随机位置替换它的一些值。

我找到了一些使用 numpy.random.choice 创建的答案 数组的掩码。不幸的是，这不会创建 原始数组的 View ，因此我无法替换它的值。

所以这是我想做的一个例子。

假设我有一个带有浮点值的二维数组。

[[ 1., 2., 3.],
 [ 4., 5., 6.],
 [ 7., 8., 9.]]

然后我想替换任意数量的 元素。如果我能用参数调整就好了 有多少元素将被替换。 可能的结果如下所示:

[[ 3.234, 2., 3.],
 [ 4., 5., 6.],
 [ 7., 8., 2.234]]

我想不出完成此任务的好方法。 感谢您的帮助。

Answer 1

只需用一个随机的相同形状的数组来掩盖您的输入数组。

import numpy as np

# input array
x = np.array([[ 1., 2., 3.], [ 4., 5., 6.], [ 7., 8., 9.]])

# random boolean mask for which values will be changed
mask = np.random.randint(0,2,size=x.shape).astype(np.bool)

# random matrix the same shape of your data
r = np.random.rand(*x.shape)*np.max(x)

# use your mask to replace values in your input array
x[mask] = r[mask]

产生这样的东西:

[[ 1.          2.          3.        ]
 [ 4.          5.          8.54749399]
 [ 7.57749917  8.          4.22590641]]