掩码 numpy 数组将 nan 评估为 True

Question

考虑一个numpy包含数据的数组：

aa = np.array([-4.793, -1.299, 0.453, np.nan, np.nan, 1.131, 0.684,  1.037])

我需要像这样创建一个面具：

mask = -4. < aa

其评估为

array([False, True, True, False, False, True, True, True], dtype=bool)

这里有一个问题：我需要nan评估为True.

我正在寻找一个不涉及修改我的输入数组的通用解决方案aa。

Answer 1

用逻辑函数很简单

import numpy as np

aa = np.array([-4.793, -1.299, 0.453, np.nan, np.nan, 1.131, 0.684,  1.037])

mask = np.logical_or(-4 < aa, np.isnan(aa))

print mask
# [False  True  True  True  True  True  True  True]