python - 用 NaN 评估

Question

我想在np.all()参考信号的帮助下比较信号。引用可以包含np.nan. 在这个 NaN 位置，信号是否满足比较条件并不重要。以下是示例代码。但是这段代码不起作用。我可以比较一个for-loop 中的每个元素，但有没有一种聪明的方法来进行这种比较？

import numpy as np

reference = np.array([np.nan, 1])
signal = np.array([2, 2])

print np.all(reference < signal)

Answer 1

使用适当的切片将比较限制为非 nan 值。您也可以使用np.finite而不是~np.isnan

import numpy as np

reference = np.array([np.nan, 1])
signal = np.array([2, 2])
idx = ~np.isnan(reference)

print np.all(reference[idx] < signal[idx])