NumPy: вычислить взвешенный для данного массива

Статистика NumPy: упражнение 6 с решением

Напишите программу NumPy для вычисления веса данного массива.

Пример решения : -

Код Python:

import numpy as np
x = np.arange(5)
print("\nOriginal array:")
print(x)
weights = np.arange(1, 6)
r1 = np.average(x, weights=weights)
r2 = (x*(weights/weights.sum())).sum()
assert np.allclose(r1, r2)
print("\nWeighted average of the said array:")
print(r1)

Пример вывода:

 Исходный массив:
[0 1 2 3 4]
Средневзвешенное значение указанного массива:
2,6666666666666665

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу NumPy для вычисления медианы сглаженного заданного массива.
Далее: Напишите программу NumPy для вычисления среднего значения, стандартного отклонения и дисперсии данного массива вдоль второй оси.