python - 更改不在索引列表中的 NumPy 数组的值

Question

我有一个 NumPy 数组，例如：

a = np.arange(30)

我知道我可以indices=[2,3,4]使用例如花哨的索引替换位于位置的值：

a[indices] = 999

但是如何替换不在的位置的值indices呢？会像下面这样吗？

a[ not in indices ] = 888

score 48 · Accepted Answer

我不知道一种干净的方法来做这样的事情：

mask = np.ones(a.shape,dtype=bool) #np.ones_like(a,dtype=bool)
mask[indices] = False
a[~mask] = 999
a[mask] = 888

当然，如果您更喜欢使用 numpy 数据类型，您可以使用dtype=np.bool_-- 输出不会有任何差异。这只是一个偏好问题。

score 6 · Accepted Answer

仅适用于一维数组：

a = np.arange(30)
indices = [2, 3, 4]

ia = np.indices(a.shape)

not_indices = np.setxor1d(ia, indices)
a[not_indices] = 888

score 4 · Accepted Answer

显然，集合没有通用not运算符。您的选择是：

从一组通用索引中减去你的indices集合（取决于的形状a），但这将有点难以实现和阅读。
某种迭代（可能for-loop 是您最好的选择，因为您肯定希望使用索引已排序的事实）。
创建一个充满新值的新数组，并有选择地从旧数组中复制索引。
```
b = np.repeat(888, a.shape)
b[indices] = a[indices]
```

score 4 · Accepted Answer

只需克服类似的情况，就可以这样解决：

a = np.arange(30)
indices=[2,3,4]

a[indices] = 999

not_in_indices = [x for x in range(len(a)) if x not in indices]

a[not_in_indices] = 888

4 回答 4