按索引雕刻二维数字阵列

Question

如何根据索引公式“雕刻”或遮罩2D numpy数组？我不关心元素值是什么，只关心它在数组中的位置。在

例如，给定一个mxm数组，如何提取地址符合的所有元素

for i in range(0,m):
    for j in range(0,m):
        if j-i-k>=0:
            A[i,j] = 1
        elif j-p-k>=0:
            A[i,j] = 1
        elif i-k>=0:
            A[i,j] = 1
        else:
            A[i,j] = 0
        j=j+1
    i=i+1

在哪里

k和p是任意的栅栏

假设

^{pr2}$

这看起来像是一个对角线部分+一个水平部分+一个垂直部分。没有上面的for循环可以完成吗？在

Answer 1

xdim,ydim = data.shape
k = 2
a, b = np.meshgrid(range(ydim),range(xdim))
mask = (b - a -k) < 0

new_data = data[mask]

new_data2 = np.array(data) # to force a copy
new_data2[~mask] = 0

new_data是一个向量，因为掩蔽过程（以这种方式完成）会使数组变平。您选择的形状参差不齐，无法用数组表示。如果只想将未选择的值设置为0，请使用新的“数据2”。在