python教程—比较包含NaN的numpy数组-Python实用宝典

python教程—比较包含NaN的numpy数组

对于我的unittest,我想检查两个数组是否相同。简化示例:这不起作用,因为nan != nan。 最好的方法是什么?

对于我的unittest,我想检查两个数组是否相同。减少的例子:

    a = np.array([1, 2, np.NaN]) b = np.array([1, 2, np.NaN]) if np.all(a==b): print 'arrays are equal'

这不起作用,因为nan != nan。
最好的方法是什么?

回答

或者,您可以使用numpy.testing。assert_equalnumpy.testing。assert_array_equal with a try/except:

    In : import numpy as np In : def nan_equal(a,b): ...: try: ...: np.testing.assert_equal(a,b) ...: except AssertionError: ...: return False ...: return True In : a=np.array([1, 2, np.NaN]) In : b=np.array([1, 2, np.NaN]) In : nan_equal(a,b) Out: True In : a=np.array([1, 2, np.NaN]) In : b=np.array([3, 2, np.NaN]) In : nan_equal(a,b) Out: False

<强>编辑< /强>

由于您将其用于单元测试,所以裸断言(而不是将其包装为True/False)可能更自然。

​Python实用宝典 (pythondict.com)
不只是一个宝典
欢迎关注公众号:Python实用宝典

本文由 Python实用宝典 作者:Python实用宝典 发表,其版权均为 Python实用宝典 所有,文章内容系作者个人观点,不代表 Python实用宝典 对观点赞同或支持。如需转载,请注明文章来源。
0

发表评论