在由切片和两个布尔数组索引的numpy数组中设置值

时间:2013-09-05 10:54:44

标签: python arrays numpy boolean

我有两个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的形状为{{1} }。

我想我在这里试图做一些相对简单的事情。如何使用这两个布尔数组进行索引,以便我可以设置数组的值?

1 个答案:

答案 0 :(得分:4)

切片numpy数组会返回一个视图,但布尔索引会返回一个数组的副本。所以当你第一次使用a[slice_][indices][high_indices]中的布尔索引索引它时,你得到了一个副本,并且值42被分配给副本而不是数组a。您可以通过链接布尔索引来解决问题:

a[slice_][(a[slice_] > 700) & (b[slice_] > 0.5)] = 42