如果列包含NaN，如何让数组表示NaN

Question

我们知道：

>>> a = np.array([[1, np.nan], [3, 4]])
>>> np.nanmean(a, axis=0)
array([ 2.,  4.])

但是如何让结果成为

array([ 2.,  np.nan])