NumPy: вычисление средних значений с удаленными значениями NaN

Как я могу вычислить средние значения матрицы по матрице, но удалить значения nan из расчета? (Для людей R подумайте na.rm = TRUE ).

Вот мой [не] рабочий пример:

import numpy as np
dat = np.array([[1, 2, 3],
                [4, 5, np.nan],
                [np.nan, 6, np.nan],
                [np.nan, np.nan, np.nan]])
print(dat)
print(dat.mean(1))  # [  2.  nan  nan  nan]

Если удалить NaN, мой ожидаемый результат будет:

array([ 2.,  4.5,  6.,  nan])
39
задан Mike T 13 February 2014 в 23:00
поделиться