首页 > 解决方案 > 如何比较 Python 中具有 np.nan 值的 2 个列表,因为 np.nan 不可比较?

问题描述

我需要使用 pytest 编写一些测试来比较包含 np.nan 值的 2 个列表(或数组)。例如,

array_expected = np.array([0, 1, 0.4, np.nan], [np.nan, 1.3, 2, 5])

def test_array(args, array_expected):
    #function that generate the array
    array_result = somefunction(args)
    assert array_result == array_expected

这将永远失败,因为np.nan != np.nan

有什么解决办法吗?谢谢

标签: pythonpytestnan

解决方案


推荐阅读