Как рассчитать средние значения матрицы по матрице, но удалить значения 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])