NumPy 中位数函数

Beginner

介绍

NumPy 是一个用于处理数组的 Python 库。它还支持对数组进行数学运算。其中一个数学运算是查找数组的中位数。中位数是一组数据的中间值,用于表示一组数字的平均值,并且不受异常值的影响。在本实验中,我们将学习如何使用 NumPy 的中位数函数。

虚拟机提示

虚拟机启动完成后,点击左上角切换到 Notebook 选项卡以访问 Jupyter Notebook 进行练习。

有时,你可能需要等待几秒钟,直到 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制,操作验证无法自动化。

如果你在学习过程中遇到问题,请随时联系 Labby。在会话结束后提供反馈,我们将及时为你解决问题。

导入 NumPy 库

在使用 NumPy 库之前,我们需要先导入它。

import numpy as np

创建一个一维数组

让我们使用 numpy.array() 方法创建一个一维数组,该数组将作为计算中位数的输入。

a = np.array([26, 2, 73, 13, 34])

计算一维数组的中位数

我们使用 numpy.median() 方法来计算数组的中位数。由于这是一个一维数组,因此不需要指定任何轴(axis)。

median = np.median(a)
print("The median of the 1D array is:", median)

输出:

The median of the 1D array is: 26.0

创建一个二维数组

现在让我们创建一个包含不同输入值的二维数组。

inp = np.array([[1, 17, 19, 33, 49], [14, 6, 87, 8, 19], [34, 2, 54, 4, 7]])

计算二维数组的中位数(axis=None)

现在我们将使用 numpy.median() 方法来计算二维数组的中位数。

axis=None 时,该方法会返回数组中所有元素的中位数。

median = np.median(inp)
print("The median of array when axis=None :", median)

输出:

The median of array when axis=None : 17.0

计算二维数组的中位数(axis=0)

axis=0 时,该方法会返回二维数组沿列方向的中位数。

median = np.median(inp, axis=0)
print("The median of array when axis=0 :", median)

输出:

The median of array when axis=0 : [14.  6. 54.  8. 19.]

计算二维数组的中位数(axis=1)

axis=1 时,该方法会返回二维数组沿行方向的中位数。

median = np.median(inp, axis=1)
print("The median of array when axis=1 :", median)

输出:

The median of array when axis=1 : [19. 14.  7.]

总结

在本实验中,我们学习了 NumPy 库以及如何使用它来计算数组的中位数。我们还介绍了 numpy.median() 函数的语法和参数。此外,我们通过示例演示了如何在一维和二维数组中计算中位数。