使用布尔掩码数组

Question

我收到关于numpy.where(mask == False)的PEP8投诉，其中mask是一个布尔数组。 PEP8建议比较应为“如果条件为假”或“如果不是条件”。 numpy.where()内建议比较的pythonic语法是什么？

Answer 1

在NumPy中取消布尔掩码数组是~mask。

另外，考虑一下你是否真的需要where。看似最常见的用途是some_array[np.where(some_mask)]，但这只是一种不必要的冗长而低效的方式来编写some_array[some_mask]。