Как позволить массиву означать NaN, если столбцы содержат NaN

Question

Мы знаем:

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

Но как сделать так, чтобы результат был

array([ 2.,  np.nan])