添加两个 2D NumPy 数组，忽略其中的 NaN

将 numpy.nan 作为缺失值添加 2 个 numpy 数组 a 和 b（均为 2D）的正确方法是什么？

1. a + b

or

2. numpy.ma.sum(a,b)

由于输入是二维数组，因此您可以将它们沿第三轴堆叠np.dstack http://docs.scipy.org/doc/numpy/reference/generated/numpy.dstack.html然后使用np.nansum http://docs.scipy.org/doc/numpy/reference/generated/numpy.nansum.html这将确保NaNs被忽略，除非有NaNs在两个输入数组中，在这种情况下输出也会有NaN。因此，实现看起来像这样 -

np.nansum(np.dstack((A,B)),2)

样本运行 -

In [157]: A
Out[157]: 
array([[ 0.77552455,  0.89241629,         nan,  0.61187474],
       [ 0.62777982,  0.80245533,         nan,  0.66320306],
       [ 0.41578442,  0.26144272,  0.90260667,         nan],
       [ 0.65122428,  0.3211213 ,  0.81634856,         nan],
       [ 0.52957704,  0.73460363,  0.16484994,  0.20701344]])

In [158]: B
Out[158]: 
array([[ 0.55809925,  0.1339353 ,         nan,  0.35154039],
       [ 0.94484722,  0.23814073,  0.36048809,  0.20412318],
       [ 0.25191484,         nan,  0.43721322,  0.95810905],
       [ 0.69115038,  0.51490958,         nan,  0.44613473],
       [ 0.01709308,  0.81771896,  0.3229837 ,  0.64013882]])

In [159]: np.nansum(np.dstack((A,B)),2)
Out[159]: 
array([[ 1.3336238 ,  1.02635159,         nan,  0.96341512],
       [ 1.57262704,  1.04059606,  0.36048809,  0.86732624],
       [ 0.66769925,  0.26144272,  1.33981989,  0.95810905],
       [ 1.34237466,  0.83603089,  0.81634856,  0.44613473],
       [ 0.54667013,  1.55232259,  0.48783363,  0.84715226]])