如何删除 3D 矩阵中的零元素？

Question

我有两个数组 input 和 label。每个数组的大小为3x2x2x2，表示2x2x2 矩阵的3 元素。数组label 有一些 元素（我的波纹管代码中的=2）的值为零。我想将它们删除/删除到 label 数组。我还想删除数组input 中对应于label 数组中已删除索引的元素。我怎样才能在python中做到这一点？谢谢

import numpy as np
input=np.arange(3*2*2*2).reshape(3,2,2,2)
label=np.arange(3*2*2*2).reshape(3,2,2,2)
# Assign value
label[0]=0
label[2]=0   
print input
print label
#Delete element 
for i in range(label.shape[0]):
    if(np.sum(label[i])==0):
        print i
        #del label[i, :, :, :]
#Expected result: non-zero element are label[1] and input[1]

Answer 1

一种掩蔽方法 -

mask = ~(label == 0).all(axis=(1,2,3)) # or (label != 0).any(axis=(1,2,3))
new_label = label[mask]
new_input = input[mask]

我将这些label[mask] 和input[mask] 输出为新的输出数组，只是为了显示/指示这些数组是新数组而不是输入数组的视图。严格来说，从数组中删除元素是不可能的，而是重新分配。所以，我们可以做重新分配的部分，像这样 -

label = label[mask]
input = input[mask]

---

另一种方法是沿第一个轴查找这些索引，然后使用np.delete -

idx = np.flatnonzero((label == 0).all((1,2,3)))
new_label = np.delete(label, idx, axis=0)
new_input = np.delete(input, idx, axis=0)

同样，我们实际上并没有删除元素，而只是创建了新的副本，可以像之前讨论的那样重新分配。