要计算加权平均值,可以使用numpy库的dot函数,该函数可以实现向量的点积运算。下面是一个示例代码,演示如何计算加权平均值:
import numpy as np
# 定义数据和权重
data = np.array([1, 2, 3, 4, 5])
weights = np.array([0.1, 0.2, 0.3, 0.2, 0.2])
# 计算加权平均值
weighted_avg = np.dot(data, weights)
print("加权平均值:", weighted_avg)
运行结果:
加权平均值: 2.9
在这个例子中,数据是一个包含5个元素的一维数组,而权重也是一个包含5个元素的一维数组。通过使用np.dot函数,可以将数据和权重进行点积运算,得到加权平均值。