Функция var
Функция var вычисляет дисперсию элементов массива.
Первым параметром функция принимает массив, для которого
необходимо вычислить дисперсию. Вторым параметром можно
задать ось, вдоль которой производится вычисление.
Третьим параметром указывается степень свободы для
поправки на смещение. Дополнительно можно передать
параметр dtype для указания типа данных,
а также параметр keepdims для сохранения
размерности массива.
Синтаксис
np.var(a, [axis], [dtype], [out], [ddof], [keepdims])
Пример
Вычислим дисперсию одномерного массива чисел:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res = np.var(arr)
print(res)
Результат выполнения кода:
2.0
Пример
Вычислим дисперсию двумерного массива вдоль разных осей:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
res_axis0 = np.var(arr, axis=0)
res_axis1 = np.var(arr, axis=1)
print(res_axis0)
print(res_axis1)
Результат выполнения кода:
[2.25 2.25 2.25]
[0.66666667 0.66666667]
Пример
Вычислим дисперсию с использованием параметра ddof
для несмещённой оценки (выборочная дисперсия):
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res_ddof0 = np.var(arr, ddof=0)
res_ddof1 = np.var(arr, ddof=1)
print(res_ddof0)
print(res_ddof1)
Результат выполнения кода:
2.0
2.5
Пример
Вычислим дисперсию массива с сохранением размерности
с помощью параметра keepdims:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.var(arr, axis=1, keepdims=True)
print(res)
print(res.shape)
Результат выполнения кода:
[[0.66666667]
[0.66666667]]
(2, 1)