欢迎访问宙启技术站
智能推送

Python函数库介绍:NumPy基础操作

发布时间:2023-06-14 13:27:34

NumPy是Python语言的一个数学库,它提供了一个强大的数组对象和许多数学函数,用于处理多维数组。NumPy可以说是科学计算中最重要的一个库了,几乎所有的科学计算都需要用到NumPy来处理数据。

NumPy的核心是多维数组numpy.ndarray,它是一个由同类数据元素构成的矩阵。在NumPy中,维度叫做轴(axis),轴的个数叫做秩(rank)。例如,一个一维数组就是一个秩为1的数组,而一个矩阵是一个秩为2的数组。例如:

import numpy as np

a = np.array([1, 2, 3]) # 一维数组
print(a) # 输出 [1 2 3]
print(a.ndim) # 输出数组的维度,即秩,为1

b = np.array([[1, 2], [3, 4]]) # 二维数组
print(b) # 输出 [[1 2] [3 4]]
print(b.ndim) # 输出数组的维度,即秩,为2

NumPy中数组的索引是从0开始的,与Python列表的索引类似。NumPy也可以进行切片,切片语法与Python的切片语法相同。

import numpy as np

a = np.array([1, 2, 3, 4, 5])
print(a[1]) # 输出 2
print(a[1:3]) # 输出 [2 3]
print(a[1:-1]) # 输出 [2 3 4]

b = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(b[0, 1]) # 输出 2
print(b[:2, :2]) # 输出 [[1 2] [4 5]]
print(b[1:, 1:]) # 输出 [[5 6] [8 9]]

NumPy还提供了许多的数学函数,例如:

import numpy as np

a = np.array([1, 2, 3])
print(np.sum(a)) # 输出数组元素的总和,即6
print(np.mean(a)) # 输出数组元素的平均值,即2
print(np.std(a)) # 输出数组元素的标准差,即0.816496580927726
print(np.min(a)) # 输出数组元素的最小值,即1
print(np.max(a)) # 输出数组元素的最大值,即3

b = np.array([[1, 2], [3, 4]])
print(np.sum(b, axis=0)) # 输出列相加的结果,即[4 6]
print(np.sum(b, axis=1)) # 输出行相加的结果,即[3 7]

当然,NumPy的功能远远不止这些,它还支持多维数组的运算、线性代数和随机数生成等操作。需要深入了解NumPy的功能,建议通过参考官方指南进行学习。

总之,NumPy提供了一个高效的算法库,能够方便地处理多维数组和数学函数,使得科学计算变得简单而又优雅。NumPy是每个Python程序员都应该了解和熟练掌握的一个工具。