小编Hok*_*rra的帖子

如何根据索引公式"雕刻"或掩盖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是任意围栏

假设

k<m
p<m

这最终看起来像对角切片+水平切片+垂直切片.可以在没有上面的for循环的情况下完成吗？