有没有一种惯用的方法来比较两个将NaN视为相等(但不等于除NaN之外的任何值)的NumPy数组。
例如,我希望下面两个数组比较相等:
np.array([1.0, np.NAN, 2.0])
np.array([1.0, np.NAN, 2.0])
与以下两个数组比较不相等:
np.array([1.0, np.NAN, 2.0])
np.array([1.0, 0.0, 2.0])
我正在寻找一种能产生标量布尔结果的方法。
下面的代码可以做到这一点:
np.all((a == b) | (np.isnan(a) & np.isnan(b)))
但它很笨拙,而且创建了所有这些中间数组。
有没有一种方法可以让眼睛看起来更舒服,并更好地利用内存?
附注:如果有帮助,我们知道这些数组具有相同的形状和数据类型。
https://stackoverflow.com/questions/10819715
复制相似问题