计算给定 NumPy 数组的加权平均值

当我们需要计算数组中某些元素的平均值时，可以使用NumPy模块中的mean()函数。然而，如果我们想在计算平均值的过程中给不同的元素设置不同的权重，则需要使用加权平均值。在本文中，我们将介绍如何计算给定NumPy数组的加权平均值。

加权平均值的公式

首先，我们需要了解加权平均值的公式。加权平均值可以表示为以下公式：

加权平均值 = (w1x1 + w2x2 + ... + wn*xn) / (w1+w2+...+wn)

其中，x1, x2, ..., xn是要计算平均值的元素， w1, w2, ..., wn是相应元素的权重。

计算加权平均值的方法

现在，让我们看看如何使用NumPy模块计算给定数组的加权平均值。假设我们有以下数组：

import numpy as np

x = np.array([1, 2, 3, 4, 5])
weights = np.array([0.1, 0.2, 0.3, 0.2, 0.2])

其中，数组x中包含要计算平均值的元素，数组weights包含相应元素的权重。

我们可以使用NumPy中的dot()函数将数组x和weights相乘，并将结果相加。然后，我们可以将结果除以权重数组的总和，得到加权平均值，如下所示：

weighted_average = np.sum(x * weights) / np.sum(weights)

在这个例子中，加权平均值为2.9。您可以将上述代码放入程序中，并根据需要更改输入数组和权重，计算该数组的加权平均值。

总结

本文介绍了计算NumPy数组的加权平均值的方法。首先，我们了解了加权平均值的公式，然后使用NumPy中的dot()函数和sum()函数计算加权平均值。加权平均值对于对不同元素分配不同权重的情况非常有用。