NumPy(Numerical Python)是Python科学计算的核心库。它提供了高性能的多维数组对象(ndarray)以及用于快速操作这些数组的工具。
NumPy是开源的Python库,几乎应用于所有科学计算和数据分析领域。它的主要功能包括:
ndarray,支持矢量化运算,避免了Python原生循环的低效。NumPy是许多高级扩展库的基础,例如SciPy(科学计算)、Pandas(数据分析)、Matplotlib(数据可视化)和scikit-learn(机器学习)等。
使用pip可以轻松安装NumPy(推荐在虚拟环境中进行):
pip install numpy
如果你使用Anaconda发行版,NumPy通常已经预装。也可以通过conda安装:
conda install numpy
下面是一个简单的示例,展示如何创建NumPy数组并进行基本运算:
import numpy as np
# 从列表创建一维数组
arr1 = np.array([1, 2, 3, 4, 5])
print("一维数组:", arr1)
# 创建二维数组
arr2 = np.array([[1, 2, 3], [4, 5, 6]])
print("二维数组:\n", arr2)
# 数组运算:标量乘法
print("arr1 * 2 =", arr1 * 2)
# 数组运算:逐元素加法
print("arr2 + 10:\n", arr2 + 10)
# 数组形状信息
print("arr2的形状:", arr2.shape)
print("arr2的数据类型:", arr2.dtype)
输出结果:
一维数组: [1 2 3 4 5]
二维数组:
[[1 2 3]
[4 5 6]]
arr1 * 2 = [ 2 4 6 8 10]
arr2 + 10:
[[11 12 13]
[14 15 16]]
arr2的形状: (2, 3)
arr2的数据类型: int64
与Python原生的列表相比,NumPy数组具有以下优势:
无论你是数据科学家、机器学习工程师还是科研人员,NumPy都是你工具箱中不可或缺的一员。