我有两个 numpy 数组:
a = np.arange(100*100).reshape(100,100)
b = np.random.rand(100, 100)
我还有一个切片元组来提取数组的某个部分:
slice_ = (slice(5, 10), slice(5, 10))
然后我有一组布尔索引来选择这个切片的某个部分:
indices = b[slice_] > 0.5
如果我想将这些索引设置为不同的值,我可以轻松做到:
a[slice_][indices] = 42
但是,如果我创建另一组布尔索引来选择索引数组的特定部分:
high_indices = a[slice_][indices] > 700
然后尝试在这些索引处设置数组的值:
a[slice_][indices][high_indices] = 42 # Doesn't do anything!
我想也许我需要将两个索引数组合并在一起,但它们的形状不同:indices
形状为(5, 5)
,high_indices
形状为(12,)
。
我想我在这里让自己陷入了一个可怕的混乱中,试图做一些相对简单的事情。如何以可以设置数组值的方式使用这两个布尔数组进行索引?