NumPy是Python中一个非常常用的科学计算模块，它可以帮助我们更加高效地操作多维数组并进行各种数学运算。本文将介绍NumPy的基本操作、数组运算、文件操作、广播等相关知识。
一、NumPy基础操作
NumPy中最基本的对象是ndarray对象，它是一个n维数组，可以容纳同类型的元素，每个元素占据4个或8个字节的内存空间，它的形状由一个由n个非负整数组成的元组定义。在NumPy中，ndarray对象还可以表示为dtype对象，可以指定数据类型、内存空间布局、字节顺序等属性。
1. 创建ndarray对象。可以使用NumPy函数numpy.array()来创建数组。传递一个列表或元组即可：
import numpy as np
a = np.array([1,2,3,4,5])
print(a)
输出结果：
[1 2 3 4 5]
也可以传递一个嵌套列表：
b = np.array([[1,2,3],[4,5,6]])
print(b)
输出结果：
[[1 2 3]
[4 5 6]]
2. 数组属性。可以使用ndarray对象的属性来查看数组的维度、形状、大小、元素类型等：
c = np.array([1,2,3], dtype=np.float32)
print('c的维度为：', c.ndim)
print('c的形状为：', c.shape)
print('c的大小为：', c.size)
print('c中的元素类型为：', c.dtype)
输出结果：
c的维度为： 1
c的形状为： (3,)
c的大小为： 3
c中的元素类型为： float32
二、数组运算
NumPy中的矢量化操作可以使得代码更加简洁高效，同时也可以避免循环带来的性能损耗。下面介绍常用的数组运算：
1. 数组索引和切片。可以使用整数或切片来访问ndarray对象的元素，当使用多个整数或切片时，可以用英文逗号隔开：
d = np.array([[1,2,3],[4,5,6],[7,8,9]])
print(d[0,2])
print(d[1:3,0:2])
输出结果：
3
[[4 5]
[7 8]]
2. 数组运算。NumPy可以实现数组之间的数学运算，包括加、乘、除、幂等运算。
e = np.array([1,2,3])
f = np.array([4,5,6])
print('e+f:', e+f)
print('e*f:', e*f)
print('e/f:', e/f)
print('e**2:', e**2)
输出结果：
e+f: [5 7 9]
e*f: [ 4 10 18]
e/f: [0.25 0.4 0.5 ]
e**2: [1 4 9]
3. 数学函数。NumPy中还有许多可以对数组进行数学计算的函数，例如：sin、cos、exp、sqrt等。
g = np.array([1,2,3])
print('sin(g):', np.sin(g))
print('cos(g):', np.cos(g))
print('exp(g):', np.exp(g))
print('sqrt(g):', np.sqrt(g))
输出结果：
sin(g): [0.84147098 0.90929743 0.14112001]
cos(g): [ 0.54030231 -0.41614684 -0.9899925 ]
exp(g): [ 2.71828183 7.3890561 20.08553692]
sqrt(g): [1. 1.41421356 1.73205081]
三、NumPy文件操作
NumPy可以读取和写入各种文件格式的数组。当使用NumPy保存数组时，它会自动添加一个文件头，其中包含数组的形状、类型和其他信息。
1. 保存数组。可以使用numpy.save()函数将NumPy数组保存到文件中。文件扩展名为.npy。
h = np.array([1,2,3])
np.save('h_array',h)
2. 加载数组。可以使用numpy.load()函数读取.npy文件：
i = np.load('h_array.npy')
print(i)
输出结果：
[1 2 3]
四、广播
广播是一个强大的概念，它允许NumPy在执行算术运算期间使用不同形状的数组。在许多情况下，广播可以避免使用循环，从而大大提高代码的性能。
1. 一维数组广播。当一个数组和一个标量进行运算时，标量会被广播到数组的每个元素上。
j = np.array([1,2,3])
print(j+2)
输出结果：
[3 4 5]
2. 二维数组广播。当两个数组的形状不同时，NumPy会自动执行广播。将每个数组扩展到公共形状，然后进行逐个元素的运算。例如，当一个二维数组和一个一维数组相加时，可以将一维数组扩展为二维数组的形状。
k = np.array([[1,2,3],[4,5,6]])
l = np.array([1,2,3])
print(k+l)
输出结果：
[[2 4 6]
[5 7 9]]
以上就是NumPy的基本操作、数组运算、文件操作、广播等相关知识的介绍。使用NumPy可以更加高效地处理数组运算，使得代码更加简洁高效，避免循环带来的性能损耗。
原创文章，作者：OLFN，如若转载，请注明出处：https://www.506064.com/n/146328.html
微信扫一扫 
支付宝扫一扫 