Python中NumPy库的相关操作

桃扇骨 2024-03-22 16:49 102阅读 0赞

![21e300825db9421aabf520b7bfb4b9bc.gif][]

**目录**

NumPy库

常用操作

创建数组

数组属性

索引和切片

数组运算

数组重塑

--------------------

## NumPy库 ##

NumPy（Numerical Python）是Python中常用的数值计算库，它提供了高性能的多维数组对象和对数组进行操作的函数。

1.多维数组对象（ndarray）

（1）NumPy最重要的对象是ndarray，它是一个具有固定大小的数组，可以包含相同类型的元素。

（2）ndarray的维度称为轴（axes），轴的个数称为秩（rank）。

（3）ndarray对象可以通过索引和切片进行访问和操作。

2.数组创建和初始化

（1）使用NumPy的array()函数可以创建一个ndarray对象。

（2）可以使用NumPy提供的函数创建特定类型的数组，如zeros()、ones()、arange()等。

（3）可以通过reshape()函数改变数组的形状。

3.数组的操作

（1）可以对数组进行基本的算术运算，如加法、减法、乘法、除法等。

（2）可以使用NumPy提供的函数进行数组的逐元素运算，如sqrt()、exp()、sin()等。

（3）可以对数组进行切片和索引操作，获取数组的子集。

4.数组的聚合和统计

（1）NumPy提供了很多聚合函数，如sum()、mean()、min()、max()等，用于对数组进行统计计算。

（2）可以使用axis参数指定在哪个轴上进行聚合操作。

5.数组的广播

（1）NumPy的广播（broadcasting）机制允许对形状不同的数组进行计算。

（2）在广播中，较小的数组会自动扩展成较大数组的形状，以便进行元素级别的操作。

6.数组的排序和搜索

（1）可以使用sort()函数对数组进行排序。

（2）可以使用searchsorted()函数在有序数组中执行二分搜索。

7.数组的读写和文件操作

（1）可以使用NumPy的loadtxt()和savetxt()函数读写文本文件中的数组数据。

（2）可以使用NumPy的load()和save()函数读写二进制文件中的数组数据。

8.数组的线性代数运算

（1）NumPy提供了一些函数用于进行线性代数运算，如dot()、inv()、det()等。

（2）可以使用NumPy提供的线性代数模块（linalg）进行更复杂的线性代数操作。

9.数组的随机数生成

（1）NumPy的random模块提供了生成随机数的函数，如rand()、randn()、randint()等。

### 常用操作 ###

#### 创建数组 ####

import numpy as np
    
    # 通过列表创建一维数组
    arr1 = np.array([1, 2, 3, 4, 5])
    
    # 通过元组创建二维数组
    arr2 = np.array([(1, 2, 3), (4, 5, 6)])
    
    # 通过范围创建一维数组
    arr3 = np.arange(1, 6)

上述代码示例中，使用NumPy库的`array`函数和`arange`函数分别创建了一维和二维数组。

#### 数组属性 ####

import numpy as np
    
    arr = np.array([1, 2, 3, 4, 5])
    
    print("数组形状：", arr.shape)
    print("数组维度：", arr.ndim)
    print("数组大小：", arr.size)

上述代码示例中，使用NumPy数组的属性`shape`、`ndim`和`size`分别获取了数组的形状、维度和大小。

#### 索引和切片 ####

import numpy as np
    
    arr = np.array([1, 2, 3, 4, 5])
    
    print("第一个元素：", arr[0])
    print("最后一个元素：", arr[-1])
    print("切片取值：", arr[1:4])

上述代码示例中，使用NumPy数组的索引和切片操作，获取了数组中的元素和部分元素。

#### 数组运算 ####

import numpy as np
    
    arr1 = np.array([1, 2, 3])
    arr2 = np.array([4, 5, 6])
    
    # 数组加法
    arr3 = arr1 + arr2
    
    # 数组乘法
    arr4 = arr1 * arr2
    
    # 数组平方
    arr5 = arr1 ** 2
    
    print("数组加法结果：", arr3)
    print("数组乘法结果：", arr4)
    print("数组平方结果：", arr5)

上述代码示例中，使用NumPy数组进行了加法、乘法和平方运算，得到了对应的结果数组。

#### 数组重塑 ####

import numpy as np
    
    arr = np.array([1, 2, 3, 4, 5, 6])
    
    # 重塑为二维数组
    arr_reshape = arr.reshape(2, 3)
    
    print("重塑后的数组：\n", arr_reshape)

上述代码示例中，使用NumPy数组的`reshape`方法将一维数组重塑为二维数组。

[21e300825db9421aabf520b7bfb4b9bc.gif]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/a563b4f659b3436e9ced0f6aa8768825.png