在 numpy 数组中跳过行

Question

我有一个带有imgs_all.shape == (511,112,112,2)的 numpy 数组imgs_all 。 我还有另一个 numpy 数组NotMissed ，大小为 511 行，用 1 和 0 填充。 每当NotMissed为零时，我将如何 go 关于跳过imgs_all中的行？ 我试图做类似以下的事情，但我认为我错了，因为我没有得到我想要的结果。

import numpy as np

N1 = SCALE - len(counter_missed) #number of rows that are not zero
imgs_all_new = np.zeros((N1,112,112,2), dtype = np.float32)

for zeros,goodimages in zip(NotMissed, imgs_all[:,:,:,:]):
     if zeros == 1:  
         imgs_all_new = goodimages
 print(imgs_all_new.shape)

Answer 1

您可以使用 boolean 数组对数组进行切片：

non_missing_imgs = NotMissed == 1
sum(non_missing_imgs) # how many True entries are there
>>> 279


imgs_all[non_missing_imgs].shape
>>> (279,112,112,2)