删除存储为数组的图像中的像素

我有一个核阵列 I 哪些商店 N 大小图像 P (像素数)。每个图像都有大小 P = q*q .

N = 1000 # number of images
q = 10 # length and width of image
P = q*q # pixels of image
I = np.ones((N,P)) # array with N images of size P

现在我想删除大小的补丁 ps 围绕选定索引 IDX (将所有值设置为零)。

ps = 2 # patch size (ps x ps)
IDX = np.random.randint(0,P,(N,1))

我的方法是使用 reshape(q,q) 删除周围的像素 小精灵 . 这里我有一个问题,就是我不知道如何计算给定图像中的位置 小精灵 . 此外,我必须检查索引是否不在图像之外。

如何解决这个问题,有没有办法把这个过程矢量化?

编辑:

在@brenla的帮助下,我做了以下的工作来移除补丁。我的方法的问题是,它需要三个for循环,并且我必须重新塑造每个图像两次。有没有办法提高性能?这一部分大大降低了我的代码速度。

import numpy as np
import matplotlib.pyplot as plt

def myplot(I):
    imgs = 10
    for i in range(imgs**2):
        plt.subplot(imgs,imgs,(i+1))
        plt.imshow(I[i].reshape(q,q), interpolation="none")
        plt.axis("off")
    plt.show()

N = 10000
q = 28
P = q*q
I = np.random.rand(N,P)

ps = 3
IDX = np.random.randint(0,P,(N,1))

for i in range(N):
    img = I[i].reshape(q,q)
    y0, x0 = np.unravel_index(IDX[i,0],(q,q))
    for x in range(ps):
        for y in range(ps):
            if (x0+x < q) and (y0+y < q):
                img[x0+x,y0+y] = 2.0
    I[i] = img.reshape(1,q*q)

myplot(I)