💡 如果想阅读最新的文章，或者有技术问题需要交流和沟通，可搜关注微信公众号号号“希望睿智”。

在上一节，我们介绍了Python的requests模块，包括：发送POST请求、发送GET请求、设置请求头、会话管理等内容。在这一节，我们将介绍Python的numpy模块。numpy模块是Python的一个非常重要的科学计算库，它提供了多维数组对象，以及一系列操作这些数组的函数。numpy还是许多科学计算库的基础，比如：SciPy、Pandas、Matplotlib等。

numpy的核心功能是多维数组对象ndarray，它是一个快速、灵活的大型数据容器，可以存储单一数据类型的元素。ndarray是同质的，即：所有元素都是相同的类型，并且可以通过索引访问。多维数组可以表示一维、二维、三维甚至更高维的数据，每个维度可以有不同的长度。

可以使用numpy.array()函数从列表或元组创建多维数组，或使用numpy.zeros()、numpy.ones()等函数创建具有指定形状和类型的多维数组。

import numpy as np# 从列表创建多维数组a = np.array([[1, 2, 3], [66, 77, 88]])# 输出：[[ 1  2  3] [66 77 88]]print(a)# 创建全部为0的多维数组b = np.zeros((2, 3))# 输出：[[0. 0. 0.] [0. 0. 0.]]print(b)  # 创建全部为1的多维数组c = np.ones((2, 4), dtype = np.int32)# 输出：[[1 1 1 1] [1 1 1 1]]print(c)

还可以使用numpy.arange()函数生成一个等差数列，类似于Python中的range()函数。该函数的语法格式为：

numpy.arange(start, stop, step, dtype)

其各个参数的含义如下。

start：起始值。

stop：终止值，生成的数列中最大的元素值，不包括该值。

step：步长，两个连续元素之间的差值。

dtype：指定生成的数组的数据类型。

import numpy as np# 生成从0到5的整数数组a = np.arange(6)# 输出：[0 1 2 3 4 5]print(a)# 生成从1开始，每次增加3，直到13的整数数组b = np.arange(1, 13, 3)# 输出：[ 1  4  7 10]print(b) # 生成从0开始，每次增加0.2，直到1的浮点数数组c = np.arange(0, 1, 0.2)# 输出：[0.  0.2 0.4 0.6 0.8]print(c)

在numpy模块中，可以使用方括号和逗号分隔的索引来访问多维数组中的元素。索引主要分为四种类型，分别为：整数索引、切片索引、布尔索引和花式索引。

整数索引：使用整数索引可以访问多维数组中的特定元素。比如：对于一个二维数组a，a[0, 0]将访问第一行第一列的元素。

import numpy as npa = np.array([[1, 2, 3], [66, 77, 88]])# 访问第一行第一列的元素，输出：1print(a[0, 0])# 访问第二行第三列的元素，输出：88print(a[1, 2])

切片索引：使用切片索引可以访问多维数组中的一部分元素。比如：对于一个二维数组a，a[0, :]将访问第一行的所有元素。

import numpy as npa = np.array([[1, 2, 3], [66, 77, 88]])# 访问第二行的所有元素，输出：[66 77 88]print(a[1, :])

布尔索引：使用布尔索引可以根据条件选择多维数组中的某些元素。比如 ：对于一个二维数组a，a[a > 5]将选择所有大于5的元素。

import numpy as npa = np.array([[1, 2, 3], [66, 77, 88]])# 访问大于2的所有元素，输出：[ 3 66 77 88]print(a[a > 2])

花式索引：使用花式索引可以根据指定的索引数组选择多维数组中的元素。比如：对于一个二维数组a，a[[0, 1], [0, 2]]将选择第一行第一列和第二行第三列的元素（第一个中括号里为元素行号，第二个中括号里为元素列号，两个括号里的元素个数必须相同）。

import numpy as npa = np.array([[1, 2, 3], [66, 77, 88]])# 输出：[ 1 88 77]print(a[[0, 1, 1], [0, 2, 1]])

numpy模块提供了大量的数组操作函数，比如：numpy.shape、numpy.reshape()、numpy.transpose()等，可以很方便地进行数组形状的改变、转置等操作。

numpy.shape属性可用于获取多维数组的形状，numpy.reshape()函数可用于改变多维数组的形状，numpy.size属性可用于获取多维数组中元素的个数。

import numpy as npa = np.array([[1, 2, 3], [66, 77, 88]])# 输出：(2, 3)print(a.shape)# 输出：6print(a.size)  b = a.reshape(3, 2)# 输出：[[ 1  2] [ 3 66] [77 88]]print(b)# 输出：6print(b.size)

numpy.append()和numpy.insert()函数可用于在数组中添加元素，numpy.delete()函数可用于从数组中删除元素，numpy.where()函数可用于返回满足指定条件的元素的索引。

import numpy as nparr = np.array([1, 2, 3])new_arr = np.append(arr, 100)# 在尾部添加数据，输出：[  1   2   3 100]print(new_arr)new_arr = np.insert(arr, 1, 100)# 在指定位置插入数据，输出：[  1 100   2   3]print(new_arr)arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])# axis默认为None，会将数组展平到一维，并删除第二个元素，输出：[1 3 4 5 6 7 8 9]new_arr = np.delete(arr, 1)print(new_arr)# 删除第二行，输出：[[1 2 3] [7 8 9]]new_arr = np.delete(arr, 1, axis = 0)print(new_arr)# 删除第二列，输出：[[1 3] [4 6] [7 9]]new_arr = np.delete(arr, 1, axis = 1)print(new_arr)arr = np.array([10, 20, 30, 40, 50])indexes = np.where(arr > 20)# 返回满足条件的索引，输出：(array([2, 3, 4], dtype=int64),)print(indexes)new_arr = np.where(arr > 20, 0, arr)# 将满足条件的元素替换为0，不满足的元素保持不变，输出：[10 20  0  0  0]print(new_arr)

numpy.sort()函数可用于对数组进行排序，numpy.concatenate()函数可用于连接多个数组。

import numpy as nparr = np.array([3, 1, 4, 1, 5, 9, 2, 6])new_arr = np.sort(arr)# 默认升序排列，输出：[1 1 2 3 4 5 6 9]print(new_arr)arr = np.array([[9, 8, 7], [6, 5, 4], [3, 2, 1]])new_arr = np.sort(arr, axis = 0)# 输出：[[3 2 1] [6 5 4] [9 8 7]]print(new_arr)new_arr = np.sort(arr, axis = 1)# 输出：[[7 8 9] [4 5 6] [1 2 3]]print(new_arr)arr1 = np.array([1, 2, 3])arr2 = np.array([4, 5, 6])new_arr = np.concatenate((arr1, arr2))# 合并数组，输出：[1 2 3 4 5 6]print(new_arr)

numpy.split(array, indices_or_sections, axis=0)函数可将数组按照指定位置或数量分割成多个子数组。indices_or_sections参数为一个整数或一个表示要分割的子数组数量的元组。如果它是一个整数，则表示将数组分割成相等大小的子数组（除了最后一个可能较小）。如果它是一个元组，则表示每个子数组的大小。例如，如果输入 (3, 2, 1)，则表示第一个子数组大小为3，第二个子数组大小为2，第三个子数组大小为1。

import numpy as nparray = np.arange(6)# 将数组分割成3个子数组，每个子数组大小相等，输出如下：# Array 1: [0 1]# Array 2: [2 3]# Array 3: [4 5]subarrays = np.split(array, 3)for i, subarray in enumerate(subarrays):    print(f"Array {i+1}:", subarray)# 将数组分割成2个子数组，第一个子数组包含2个元素，输出如下：# Array 1: [0 1]# Array 2: [2 3 4 5]indices = (2, )subarrays = np.split(array, indices)for i, subarray in enumerate(subarrays):    print(f"Array {i+1}:", subarray)

数学函数：numpy提供了大量的数学函数，可以对数组中的元素进行各种数学运算，如numpy.sin()、numpy.cos()、numpy.exp()等。

import numpy as np# 计算60度的余弦值angle = np.array([60])cos_value = np.cos(angle * np.pi / 180)# 输出：[0.5]print(cos_value)  sequence = np.array([0, 30, 45, 60, 90])sin_sequence = np.sin(sequence * np.pi / 180)# 输出：[0.         0.5        0.70710678 0.8660254  1.        ]print(sin_sequence)

线性代数：numpy还提供了一些线性代数函数，如numpy.dot()、numpy.linalg.inv()等，可以方便地进行矩阵的点积、逆矩阵等计算。

import numpy as np  a = np.array([1, 2, 3])  b = np.array([4, 5, 6])  result = np.dot(a, b)# 计算点积，输出：32print(result) a = np.array([[1, 2], [3, 4]])b = np.array([[5, 6], [7, 8]])result = np.dot(a, b)# 计算二维数组的点积，输出：[[19 22] [43 50]]print(result)

随机数生成：numpy包含了大量的随机数生成函数，可以方便地生成各种分布的随机数。

import numpy as np  random_float = np.random.rand()# 生成[0, 1)范围内的随机浮点数，输出：0.58806166392493print(random_float)random_integer = np.random.randint(0, 10)# 生成[0, 10)范围内的随机整数，输出：9print(random_integer)random_array = np.random.rand(3)# 生成长度为3的随机浮点数数组，数组中的数在[0, 1)范围内，输出：[0.90723285 0.05028758 0.13100614]print(random_array)random_array = np.random.randint(0, 100, size = 3)# 生成长度为3的随机整数数组，数组中的数在[0, 100)范围内，输出：[59 43 90]print(random_array)