淘汰multidim numpy数组的部分

Question

我有一个尺寸为（200,200,3）的numpy数组。这是一个RGB图像。

我还有这个图像区域的（xmin，ymin，xmax，ymax）坐标，我想将其设置为零。在所有三个频道中，该区域应为零。

我当然可以通过循环来解决这个问题，但这样做会很浪费。

有没有一种简单的方法可以使用numpy来掩盖数组？

Answer 1

使用数组切片。如果xmin，xmax，ymin和ymax是要设置为零的数组区域的索引，则：

a[xmin:xmax,ymin:ymax,:] = 0.