python - 如何获得 numpy 掩码数组中的掩码行数？

Question

所以我有一个 numpy 数组，其中包含许多 numpy 数组，其中一些数组具有如下所示的屏蔽值:

[[1 2 3]
 [-- -- --]
 [7 8 9]]

在更大的 numpy 数组(在本例中为 1)中获取屏蔽 numpy 数组(意思是 [-- -- --] 之类的东西)的数量的最有效方法是什么。

谢谢!

Answer 1

Masked arrays have a .mask attribute由一个 bool 数组组成，该数组在屏蔽值的任何地方都为 True。如果您想知道有多少行包含至少一个屏蔽值，您可以使用:

x.mask.any(axis=1).sum()

其中 x 是您的掩码数组。如果您只对所有 值被屏蔽的行感兴趣，您可以使用:

x.mask.all(axis=1).sum()

很明显，在您的示例中，这些结果都为 1。