0

0

如何使用NumPy进行数组计算?

紅蓮之龍

紅蓮之龍

发布时间:2025-09-06 09:41:01

|

742人浏览过

|

来源于php中文网

原创

NumPy通过提供高性能的多维数组对象和丰富的数学函数,简化了Python中的数值计算。它支持高效的数组创建、基本算术运算、矩阵乘法、通用函数及聚合操作,并具备优于Python列表的同质性、连续内存存储和底层C实现带来的性能优势。其强大的索引、切片、形状操作和广播机制进一步提升了数据处理效率,使NumPy成为科学计算的核心工具。

如何使用numpy进行数组计算?

NumPy是Python进行数值计算的核心库,它通过提供高性能的多维数组对象(ndarray)以及大量用于处理这些数组的函数,极大地简化了复杂的数学运算。它将原本需要复杂循环和低效率操作的向量、矩阵计算,转化为简洁、高效的函数调用,让Python在科学计算领域的表现力与C/C++等语言不相上下。

解决方案

要开始使用NumPy进行数组计算,首先需要导入这个库,通常我们约定俗成地将其命名为

np

import numpy as np

创建数组: NumPy数组的创建方式多种多样。你可以从Python列表或元组创建,也可以直接生成特定形状的数组,比如全零、全一、等差数列或随机数。

# 从Python列表创建一维数组
arr1 = np.array([1, 2, 3, 4, 5])
print("一维数组:", arr1)

# 从嵌套列表创建二维数组(矩阵)
arr2 = np.array([[1, 2, 3], [4, 5, 6]])
print("二维数组:\n", arr2)

# 创建全零数组
zeros_arr = np.zeros((2, 3)) # 2行3列
print("全零数组:\n", zeros_arr)

# 创建全一数组
ones_arr = np.ones((3, 2)) # 3行2列
print("全一数组:\n", ones_arr)

# 创建等差数列
range_arr = np.arange(0, 10, 2) # 从0到10(不包含10),步长为2
print("等差数列:", range_arr)

# 创建等间隔的数字
linspace_arr = np.linspace(0, 1, 5) # 从0到1,包含1,生成5个点
print("等间隔数组:", linspace_arr)

基本算术操作: NumPy数组支持元素级的算术运算,这与Python列表的行为截然不同。当两个形状相同的数组进行加减乘除时,操作会作用于对应位置的每个元素。

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

print("a + b =", a + b) # [5 7 9]
print("a - b =", a - b) # [-3 -3 -3]
print("a * b =", a * b) # [ 4 10 18] (元素级乘法)
print("a / b =", a / b) # [0.25 0.4  0.5 ]

# 标量运算
print("a + 10 =", a + 10) # [11 12 13]

矩阵乘法: 对于矩阵乘法,NumPy提供了专门的运算符

@
或函数
np.dot()

mat1 = np.array([[1, 2], [3, 4]])
mat2 = np.array([[5, 6], [7, 8]])

# 矩阵乘法
product_mat = mat1 @ mat2
print("矩阵乘法结果:\n", product_mat)
# 也可以使用 np.dot(mat1, mat2)

通用函数 (Universal Functions, ufuncs): NumPy提供了大量的ufuncs,它们是作用于

ndarray
对象的元素级数学函数,比如
sin
,
cos
,
exp
,
log
,
sqrt
等,它们都是高度优化的C实现。

data = np.array([0, np.pi/2, np.pi])
print("sin(data) =", np.sin(data))

data_exp = np.array([1, 2, 3])
print("exp(data_exp) =", np.exp(data_exp))

聚合操作: 对数组进行求和、求平均、最大值、最小值等操作也非常常见。这些操作可以作用于整个数组,也可以沿着特定轴(行或列)进行。

matrix = np.array([[1, 2, 3], [4, 5, 6]])

print("所有元素的和:", np.sum(matrix))
print("每列的和:", np.sum(matrix, axis=0)) # axis=0表示沿着列操作
print("每行的和:", np.sum(matrix, axis=1)) # axis=1表示沿着行操作

print("平均值:", np.mean(matrix))
print("最大值:", np.max(matrix))

我个人觉得,NumPy最迷人的地方在于它把那些数学上看起来很复杂的向量和矩阵运算,用几行Python代码就能优雅地解决掉,而且速度还快得惊人。这不仅仅是代码量的减少,更是思维模式的转变,从关注单个元素到关注整个数组,也就是所谓的“向量化”。

NumPy数组与Python列表有何本质区别,为何它是科学计算的首选?

NumPy数组和Python列表虽然都能存储数据集合,但它们的内部机制和设计哲学有着根本性的差异,这些差异直接决定了NumPy在科学计算领域的不可替代性。

首先,数据类型统一性是NumPy数组的一个核心特征。一个NumPy数组中的所有元素都必须是相同的数据类型(例如,都是整数、浮点数或布尔值)。而Python列表则可以存储任意类型的对象,甚至是不同类型的混合。这种同质性使得NumPy能够将数据紧密地存储在内存中的一块连续区域,这对于CPU缓存的利用和内存访问效率至关重要。相比之下,Python列表存储的是指向实际对象的指针,这些对象可能分散在内存的各个角落,访问效率自然大打折扣。

其次,性能上的巨大优势是选择NumPy的关键。NumPy的底层实现是用C和Fortran等编译型语言编写的,这意味着它的核心操作在执行时避免了Python解释器的开销。当你在NumPy数组上执行加法、乘法等操作时,实际上是在调用高度优化的C函数,而不是在Python层面对每个元素进行循环。这种“向量化”操作比Python原生的循环要快上几十甚至几百倍,尤其是在处理大规模数据集时,这种性能差异是决定性的。我刚开始用Python时,总觉得列表挺好用,直到遇到大数据量的计算瓶颈,NumPy就像是为这些场景量身定制的,那种速度提升是实实在在的。

再者,功能丰富性也是NumPy独步天下的原因。它不仅提供了多维数组对象,还集成了大量的数学函数、线性代数例程、傅里叶变换、随机数生成等高级功能。这些功能都是为高效处理数值数据而设计的,并且与整个Python科学计算生态系统(如SciPy、Pandas、Matplotlib、Scikit-learn等)无缝集成,共同构成了强大的数据分析和机器学习工具链。可以说,NumPy是这个生态系统的基石,没有它,许多高级库的效率和功能都将大打折扣。

总结来说,NumPy通过其同质化、内存连续存储的数组结构,结合底层C语言实现带来的向量化计算能力,以及其丰富的内置功能,解决了Python在处理大规模数值计算时的效率瓶痛,使其成为科学计算领域的首选工具。

掌握NumPy多维数组的索引、切片与形状操作:高效数据访问与重塑

在NumPy中,高效地访问和操作多维数组的数据是进行复杂计算的基础。理解其索引、切片和形状操作的机制,能让你像雕塑家对待泥土一样,灵活地重塑数据以适应各种分析需求。

索引与切片: NumPy的索引和切片功能远比Python列表强大,因为它支持多维操作。

  • 基本索引与切片: 你可以通过逗号分隔的索引或切片来访问多维数组的特定元素、行、列或子数组。

    arr = np.array([[1, 2, 3, 4],
                    [5, 6, 7, 8],
                    [9, 10, 11, 12]])
    
    print("访问单个元素 (第1行第2列):", arr[0, 1]) # 输出 2
    print("访问第1行:", arr[0, :]) # 输出 [1 2 3 4]
    print("访问第2列:", arr[:, 1]) # 输出 [ 2  6 10]
    print("切片子数组 (前两行,后两列):\n", arr[:2, 2:]) # 输出 [[3 4], [7 8]]
  • 整数数组索引: 这是一个非常强大的特性,允许你使用整数数组作为索引来选择非连续的元素或行/列。

    ASP.NET 4.0电子商城
    ASP.NET 4.0电子商城

    在现实生活中的购物过程,购物者需要先到商场,找到指定的产品柜台下,查看产品实体以及标价信息,如果产品合适,就将该产品放到购物车中,到收款处付款结算。电子商务网站通过虚拟网页的形式在计算机上摸拟了整个过程,首先电子商务设计人员将产品信息分类显示在网页上,用户查看网页上的产品信息,当用户看到了中意的产品后,可以将该产品添加到购物车,最后使用网上支付工具进行结算,而货物将由公司通过快递等方式发送给购物者

    下载
    # 选择第0行和第2行
    print("选择特定行:\n", arr[[0, 2]])
    # 选择 (0,1) 和 (2,3) 处的元素
    print("选择特定不连续元素:", arr[[0, 2], [1, 3]]) # 输出 [ 2 12]
  • 布尔索引: 这是数据过滤的利器。你可以创建一个与数组形状相同的布尔数组,其中

    True
    对应你想选择的元素,
    False
    则相反。

    print("大于5的元素:", arr[arr > 5]) # 输出 [ 6  7  8  9 10 11 12]
  • 视图与副本: NumPy切片通常返回的是原数组的视图(view),这意味着修改视图也会修改原数组。如果你需要独立的数据副本,务必使用

    .copy()
    方法。这是一个常见的“坑”,我记得有一次调试一个模型,数据维度总是对不上,后来才发现是
    reshape
    transpose
    没用好,一旦理解了这些,代码就顺畅多了。

形状操作: 改变数组的形状而不改变其数据是数据预处理和模型输入调整的常见操作。

  • reshape()
    改变数组的维度。
    -1
    是一个非常方便的占位符,NumPy会自动计算该维度的大小。

    arr_flat = np.arange(1, 10) # [1 2 3 4 5 6 7 8 9]
    print("重塑为3x3矩阵:\n", arr_flat.reshape(3, 3))
    print("重塑为3行,列数自动计算:\n", arr_flat.reshape(3, -1))
  • transpose()
    (
    .T
    ):
    对数组进行转置,对于二维数组就是行和列的互换,对于多维数组则是轴的重新排列

    matrix = np.array([[1, 2], [3, 4]])
    print("原矩阵:\n", matrix)
    print("转置矩阵:\n", matrix.T)
  • np.newaxis
    None
    增加数组的维度,常用于广播操作前的准备。

    vec = np.array([1, 2, 3])
    print("原向量形状:", vec.shape) # (3,)
    row_vec = vec[np.newaxis, :] # 增加一个行维度
    print("行向量形状:", row_vec.shape) # (1, 3)
    col_vec = vec[:, np.newaxis] # 增加一个列维度
    print("列向量形状:", col_vec.shape) # (3, 1)
  • flatten()
    /
    ravel()
    将多维数组展平为一维数组。
    flatten()
    总是返回副本,而
    ravel()
    可能返回视图或副本,取决于具体情况,通常更推荐
    ravel()
    因为它更高效。

    multi_dim_arr = np.array([[1, 2], [3, 4]])
    print("展平为一维数组 (flatten):", multi_dim_arr.flatten())
    print("展平为一维数组 (ravel):", multi_dim_arr.ravel())

    掌握这些操作,你就能自如地在NumPy中处理各种数据结构,为后续的数据分析和算法实现打下坚实基础。

优化NumPy数组计算:避免性能瓶颈与常见错误

虽然NumPy本身以高性能著称,但在实际使用中,如果不注意一些细节,仍然可能陷入性能陷阱。理解并避免这些常见错误,能让你的代码运行得更快、更有效率。

1. 向量化操作优先,避免Python循环: 这是NumPy性能优化的黄金法则。NumPy的优势在于其底层C实现对整个数组进行操作(向量化)。如果你在NumPy数组上使用Python原生的

for
循环来逐个元素处理,那么你几乎完全放弃了NumPy带来的性能提升,因为每次循环迭代都会产生Python解释器的开销。

# 避免这种做法 (慢)
arr = np.arange(1000000)
result = np.zeros_like(arr, dtype=float)
for i in range(len(arr)):
    result[i] = np.sin(arr[i]) * 2 + 1

# 推荐这种做法 (快)
result_vec = np.sin(arr) * 2 + 1

我曾经遇到过一个问题,就是在一个循环里对NumPy数组的每一行进行复杂操作,结果慢得令人发指。后来改成用NumPy的

apply_along_axis
或者直接重构为向量化操作,性能立马就上来了。那种从几分钟到几秒的提升,真的让人印象深刻。

2. 理解并利用广播机制: 广播(Broadcasting)是NumPy的一项强大功能,它允许不同形状的数组在特定条件下进行算术运算。正确利用广播可以避免创建不必要的中间数组,从而节省内存和计算时间。如果你发现自己在手动扩展数组维度以匹配形状,很可能可以通过广播更优雅地解决。

# 假设有一个2x3的矩阵和一个1x3的行向量
matrix = np.array([[1, 2, 3], [4, 5, 6]])
row_vec = np.array([10, 20, 30])

# NumPy会自动将row_vec广播到matrix的每一行
result = matrix + row_vec
print("广播结果:\n", result)

**3

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
C语言变量命名
C语言变量命名

c语言变量名规则是:1、变量名以英文字母开头;2、变量名中的字母是区分大小写的;3、变量名不能是关键字;4、变量名中不能包含空格、标点符号和类型说明符。php中文网还提供c语言变量的相关下载、相关课程等内容,供大家免费下载使用。

401

2023.06.20

c语言入门自学零基础
c语言入门自学零基础

C语言是当代人学习及生活中的必备基础知识,应用十分广泛,本专题为大家c语言入门自学零基础的相关文章,以及相关课程,感兴趣的朋友千万不要错过了。

620

2023.07.25

c语言运算符的优先级顺序
c语言运算符的优先级顺序

c语言运算符的优先级顺序是括号运算符 > 一元运算符 > 算术运算符 > 移位运算符 > 关系运算符 > 位运算符 > 逻辑运算符 > 赋值运算符 > 逗号运算符。本专题为大家提供c语言运算符相关的各种文章、以及下载和课程。

354

2023.08.02

c语言数据结构
c语言数据结构

数据结构是指将数据按照一定的方式组织和存储的方法。它是计算机科学中的重要概念,用来描述和解决实际问题中的数据组织和处理问题。数据结构可以分为线性结构和非线性结构。线性结构包括数组、链表、堆栈和队列等,而非线性结构包括树和图等。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

259

2023.08.09

c语言random函数用法
c语言random函数用法

c语言random函数用法:1、random.random,随机生成(0,1)之间的浮点数;2、random.randint,随机生成在范围之内的整数,两个参数分别表示上限和下限;3、random.randrange,在指定范围内,按指定基数递增的集合中获得一个随机数;4、random.choice,从序列中随机抽选一个数;5、random.shuffle,随机排序。

607

2023.09.05

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

531

2023.09.20

c语言get函数的用法
c语言get函数的用法

get函数是一个用于从输入流中获取字符的函数。可以从键盘、文件或其他输入设备中读取字符,并将其存储在指定的变量中。本文介绍了get函数的用法以及一些相关的注意事项。希望这篇文章能够帮助你更好地理解和使用get函数 。

647

2023.09.20

c数组初始化的方法
c数组初始化的方法

c语言数组初始化的方法有直接赋值法、不完全初始化法、省略数组长度法和二维数组初始化法。详细介绍:1、直接赋值法,这种方法可以直接将数组的值进行初始化;2、不完全初始化法,。这种方法可以在一定程度上节省内存空间;3、省略数组长度法,这种方法可以让编译器自动计算数组的长度;4、二维数组初始化法等等。

604

2023.09.22

C++ 设计模式与软件架构
C++ 设计模式与软件架构

本专题深入讲解 C++ 中的常见设计模式与架构优化,包括单例模式、工厂模式、观察者模式、策略模式、命令模式等,结合实际案例展示如何在 C++ 项目中应用这些模式提升代码可维护性与扩展性。通过案例分析,帮助开发者掌握 如何运用设计模式构建高质量的软件架构,提升系统的灵活性与可扩展性。

14

2026.01.30

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.7万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号