如何在数量的矩阵中删除矩阵？

Question

我有一个 numpy数组 arr_seg_labs具有以下形状：（1735，128，128）。 它包含1到10之间的像素掩模，还包含零和255（背景）。 我想删除不包含给定类别标识符（9）的那些（128，128）矩阵，并保留至少包含一个9的矩阵。 我为此做了一个蒙版（horse_mask），但是我不知道如何继续此线程来过滤此numpy数组

CAT_IDX_HORSE = 9
horse_mask = arr_seg_labs == CAT_IDX_HORSE

Answer 1

IIUC您可以使用口罩并索引为：

CAT_IDX_HORSE = 9
mask = (a == CAT_IDX_HORSE ).sum((1, 2))
result = a[mask != 0]