Функция nanmean
Функция nanmean вычисляет среднее арифметическое
вдоль указанной оси, игнорируя при этом пропущенные
значения NaN. Она полезна при работе с
реальными данными, где часто встречаются пропуски.
Первым аргументом передаётся массив, вторым -
ось (если не указана, вычисляется по всему массиву),
третьим - тип возвращаемых данных.
Синтаксис
np.nanmean(a, axis=None, dtype=None)
Пример
Вычислим среднее значение для одномерного массива с пропусками:
import numpy as np
arr = np.array([1, 2, np.nan, 4, 5])
res = np.nanmean(arr)
print(res)
Результат выполнения кода:
3.0
Пример
Вычислим среднее по строкам для двумерного массива с пропусками:
import numpy as np
arr = np.array([[1, 2, np.nan],
[4, np.nan, 6],
[7, 8, 9]])
res = np.nanmean(arr, axis=1)
print(res)
Результат выполнения кода:
[1.5 5. 8. ]
Пример
Если все элементы вдоль оси являются NaN,
результатом будет NaN:
import numpy as np
arr = np.array([[np.nan, np.nan],
[1, 2]])
res = np.nanmean(arr, axis=1)
print(res)
Результат выполнения кода:
[nan 1.5]
Пример
Укажем тип данных для точности вычислений:
import numpy as np
arr = np.array([1, 2, np.nan, 4, 5])
res = np.nanmean(arr, dtype=np.float64)
print(res)
Результат выполнения кода:
3.0
