使用 NumPy 计算数组的平均值

在 Python 中，可以使用 NumPy 库来对数组进行各种数值计算。其中之一是计算数组的平均值。

安装 NumPy

如果尚未安装 NumPy，可以使用以下命令来安装：

pip install numpy

导入 NumPy

导入 NumPy 库，以便在程序中使用它：

import numpy as np

创建数组

首先，我们需要创建一个包含需要计算平均值的元素的 NumPy 数组。可以使用 np.array() 函数来创建一个一维或多维数组。

x = np.array([1, 2, 3, 4, 5])

计算平均值

使用 np.mean() 函数来计算给定数组的平均值。将数组作为参数传递给该函数，并将结果赋值给一个变量。

average = np.mean(x)

示例代码

下面是完整示例代码：

import numpy as np

# 创建数组
x = np.array([1, 2, 3, 4, 5])

# 计算平均值
average = np.mean(x)

# 打印结果
print("平均值:", average)

运行以上代码，将输出以下结果：

平均值: 3.0

这表示数组 x 的平均值为 3.0。

总结

通过使用 NumPy 库的 np.mean() 函数，我们可以轻松计算数组的平均值。NumPy 还提供了许多其他数学和统计函数，可在需要时进行使用。