比较包含NaN的numpy数组

比较包含NaN的numpy数组

或者您可以使用

numpy.testing.assert_equal

或

numpy.testing.assert_array_equal

用

try/except

：

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

编辑

由于您正在使用它进行单元测试，因此裸露

assert

（而不是将其包装成get

True/False

）可能更自然。