[英]Setting values in a numpy arrays indexed by a slice and two boolean arrays
我有两个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,)
。
我想自己陷入了一个混乱的困境,试图做一些相对简单的事情。 如何以可以设置数组值的方式使用这两个布尔数组建立索引?
切片numpy数组会返回一个视图,但是布尔索引会返回数组的副本。 因此,当您第一次使用a[slice_][indices][high_indices]
中a[slice_][indices][high_indices]
布尔索引对它进行索引时,您将获得一个副本,并且将值42分配给副本而不是数组a
。 您可以通过链接布尔索引来解决问题:
a[slice_][(a[slice_] > 700) & (b[slice_] > 0.5)] = 42
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.