Python 是一种高级编程语言,被广泛用于数据分析、机器学习、科学计算等领域。在这些领域中,我们经常需要对数组进行各种操作,例如对数组进行排序、去重、过滤等。Python 提供了丰富的函数库,可以帮助我们高效地完成这些操作。
本文将介绍如何在 Python 中使用函数操作数组,主要包括以下几个方面:
- 创建数组
在 Python 中,我们可以使用 NumPy 库创建数组。NumPy 库提供了多种创建数组的函数,例如使用 arange 函数创建连续的一维数组:
import numpy as np
arr = np.arange(10)
print(arr)
输出结果为:
[0 1 2 3 4 5 6 7 8 9]
我们也可以使用 reshape 函数将一维数组转换为多维数组:
arr = np.arange(12).reshape(3, 4)
print(arr)
输出结果为:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
- 数组排序
在 Python 中,我们可以使用 sort 函数对数组进行排序。sort 函数默认按升序排序,也可以指定参数来进行降序排序:
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
arr.sort()
print(arr)
输出结果为:
[1 1 2 3 3 4 5 5 5 6 9]
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
arr.sort()[::-1]
print(arr)
输出结果为:
[9 6 5 5 5 4 3 3 2 1 1]
- 数组去重
在 Python 中,我们可以使用 unique 函数对数组进行去重:
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
arr = np.unique(arr)
print(arr)
输出结果为:
[1 2 3 4 5 6 9]
- 数组过滤
在 Python 中,我们可以使用 where 函数对数组进行过滤。where 函数返回满足条件的元素的下标:
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
indices = np.where(arr > 3)
print(indices)
输出结果为:
(array([2, 4, 5, 7, 8, 10]),)
我们可以使用下标来获取满足条件的元素:
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
indices = np.where(arr > 3)
print(arr[indices])
输出结果为:
[4 5 9 6 5 5]
- 数组操作
在 Python 中,我们可以使用多种函数对数组进行操作。例如,我们可以使用 sum 函数计算数组元素的和:
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
print(np.sum(arr))
输出结果为:
44
我们也可以使用 mean 函数计算数组元素的平均值:
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
print(np.mean(arr))
输出结果为:
4.0
除了以上函数,NumPy 库还提供了多种函数用于对数组进行操作,例如 min、max、argmin、argmax、cumsum、cumprod 等。
综上所述,本文介绍了如何在 Python 中使用函数操作数组。通过使用 NumPy 库提供的函数,我们可以高效地对数组进行排序、去重、过滤等操作。同时,我们也可以使用多种函数对数组进行操作,例如计算数组元素的和、平均值等。这些知识点是 Python 面试中常见的考点,掌握它们对于提高自己的编程能力和面试成功率都有很大的帮助。