Функция average
Функция average возвращает среднее взвешенное элементов массива.
Первым параметром передаётся массив данных.
Вторым параметром axis можно указать ось для вычислений.
Третьим параметром weights задаются веса для элементов.
Параметр returned определяет, возвращать ли сумму весов.
Синтаксис
np.average(a, axis=None, weights=None, returned=False)
Пример
Вычислим среднее значение одномерного массива:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res = np.average(arr)
print(res)
Результат выполнения кода:
3.0
Пример
Вычислим среднее с учётом весов:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
weights = np.array([0.1, 0.2, 0.3, 0.2, 0.2])
res = np.average(arr, weights=weights)
print(res)
Результат выполнения кода:
3.2
Пример
Вычислим среднее по строкам двумерного массива:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.average(arr, axis=0)
print(res)
Результат выполнения кода:
[2.5 3.5 4.5]
Пример
Возврат среднего и суммы весов с параметром returned:
import numpy as np
arr = np.array([1, 2, 3])
weights = np.array([1, 1, 2])
res, sum_weights = np.average(arr, weights=weights, returned=True)
print(res)
print(sum_weights)
Результат выполнения кода:
2.25
4.0
Пример
Вычислим среднее по столбцам с весами:
import numpy as np
arr = np.array([[1, 2], [3, 4], [5, 6]])
weights = np.array([0.5, 1, 0.5])
res = np.average(arr, axis=0, weights=weights)
print(res)
Результат выполнения кода:
[3. 4.]
