Функция median

Функция median вычисляет медиану - среднее значение упорядоченного набора данных. Медиана устойчива к выбросам, поэтому она часто используется в статистическом анализе. Первым параметром функция принимает массив, вторым параметром - ось для вычисления, третьим - флаг сохранения размерности.

Синтаксис

np.median(a, axis=None, keepdims=False)

Параметры:

a - входной массив (list, tuple, ndarray);
axis - ось или оси, по которым вычисляется медиана. По умолчанию (None) вычисляется по всему массиву;
keepdims - если True, размерность сохраняется, что удобно для трансляции.

Пример

Вычислим медиану одномерного массива чисел:

import numpy as np

arr = np.array([10, 20, 30, 40, 50])
res = np.median(arr)
print(res)

Результат выполнения кода:

30.0

Пример

Рассмотрим случай с чётным количеством элементов: медианой будет среднее арифметическое двух средних чисел:

import numpy as np

arr = np.array([1, 3, 5, 7])
res = np.median(arr)
print(res)

Результат выполнения кода:

4.0

Пример

Вычислим медиану для каждого столбца двумерного массива, передав параметр axis=0:

import numpy as np

arr = np.array([[5, 15, 25],
				[10, 20, 30],
				[7, 14, 21]])
res = np.median(arr, axis=0)
print(res)

Результат выполнения кода:

[7. 15. 25.]

Пример

Медиана по строкам (axis=1) для того же массива:

import numpy as np

arr = np.array([[5, 15, 25],
				[10, 20, 30],
				[7, 14, 21]])
res = np.median(arr, axis=1)
print(res)

Результат выполнения кода:

[15. 20. 14.]

Пример

Используем параметр keepdims=True, чтобы сохранить размерность результата:

import numpy as np

arr = np.array([[1, 2, 3],
				[4, 5, 6]])
res = np.median(arr, axis=1, keepdims=True)
print(res)

Результат выполнения кода:

[[2.]
 [5.]]

Пример

Если массив содержит nan, медиана также будет nan. Для пропуска пропущенных значений используйте np.nanmedian:

import numpy as np

arr = np.array([1, 2, np.nan, 4, 5])
res = np.median(arr)
print(res)

Результат выполнения кода:

nan

Смотрите также

функцию mean,
которая вычисляет среднее арифметическое элементов
функцию percentile,
которая вычисляет произвольный процентиль
функцию quantile,
которая вычисляет квантиль (от 0 до 1)
функцию std,
которая вычисляет стандартное отклонение