numpy average

numpy.average是一个用于计算数组中元素的加权平均值的函数。它可以接受一个数组和一个可选的权重数组作为输入，并返回加权平均值。

使用numpy.average函数时，可以通过设置axis参数来指定沿着哪个轴计算加权平均值。如果不指定axis参数，则默认对整个数组进行计算。

下面是numpy.average函数的用法示例：

import numpy as np

# 例子1：计算一维数组的平均值
arr = np.array([1, 2, 3, 4, 5])
average = np.average(arr)
print(average)  # 输出：3.0

# 例子2：计算二维数组的平均值
arr = np.array([[1, 2], [3, 4], [5, 6]])
average = np.average(arr, axis=0)  # 沿着列计算平均值
print(average)  # 输出：[3. 4.]

# 例子3：计算带有权重的加权平均值
arr = np.array([1, 2, 3, 4, 5])
weights = np.array([0.1, 0.2, 0.3, 0.2, 0.2])
average = np.average(arr, weights=weights)
print(average)  # 输出：3.1