python - 将掩码应用于多维数组

Question

我想以正确的方式做到这一点:

data = np.array(data)
data =[
[1, 1, 2, 1],
[0, 1, 3, 2],
[0, 2, 3, 2],
[2, 4, 3, 1],
[0, 2, 1, 4],
[3, 1, 4, 1]]

这应该变成(删除以 0 开头的行):

[1, 1, 2, 1]
[2, 4, 3, 1]
[3, 1, 4, 1]

到目前为止我是这样做的:

lines = []
for i in range(0, len(data[0])):
    if data[0,i] != 0:
        lines.append(data[:,i])
lines = np.array(lines)

然后我发现了这个好方法:

mask = 1 <= data[0,:]

现在我想将该掩码应用于该数组。该掩码内容为:[True, False, False, True, False, True]。我该怎么做？

Answer 1

为什么不只是:

[ar for ar in data if ar[0] != 0]

这假设数组不为空。