python – 掩盖numpy数组,将nan评估为True

考虑一个包含数据的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的通用解决方案.

解决方法:

使用logic function非常简单

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]

标签：python,arrays,numpy,masking
来源： https://codeday.me/bug/20190622/1263098.html