如何根据索引公式“雕刻”或遮罩2D numpy数组?我不在乎元素值是什么,只是它在数组中的位置。通过索引雕刻2D numpy数组
例如,给定一个的m×m阵列,如何抽取为地址符合
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
这最终看起来像一个对角切片+水平切片+ a垂直切片。没有上面的for循环可以完成吗?
上NEW_DATA =数据[面具] @tcaswell,用我的“为”创造面具“删除”一些行和列更多的问题,在某种程度上给我留下了一个向量,而不是原始data.shape的数组。为什么?我怎样才能防止这一点? – Hokiexterra