0

0

NumPy 2D数组高效块级修改:基于视图与查找表的策略

心靈之曲

心靈之曲

发布时间:2025-11-06 13:03:09

|

994人浏览过

|

来源于php中文网

原创

NumPy 2D数组高效块级修改:基于视图与查找表的策略

本教程旨在介绍如何利用numpy的`np.lib.stride_tricks.as_strided`功能,结合查找表机制,高效地对2d数组进行2x2块级别的修改。通过创建数组视图而非复制数据,并利用高级索引进行矢量化操作,此方法显著优于传统的python循环,极大提升了处理大型数据集的性能。

引言:Python循环的性能瓶颈

在处理大型NumPy数组时,如果需要对数组中的局部小块(例如2x2的子矩阵)进行模式识别和替换操作,传统的Python循环(如使用itertools.product遍历坐标)会因为解释器开销和逐元素操作而效率低下。NumPy的核心优势在于其矢量化操作,能够将循环操作推送到C语言层面执行,从而获得显著的性能提升。本教程的目标是展示如何采用NumPy原生的、矢量化的方式来高效地实现2D数组的2x2块级修改。

核心策略:利用NumPy视图与步幅技巧

实现高效块级操作的关键在于np.lib.stride_tricks.as_strided。这个函数允许我们创建一个现有NumPy数组的“视图”,而无需复制底层数据。通过巧妙地定义视图的shape(形状)和strides(步幅),我们可以将一个2D数组解释为一个由多个2x2块组成的更高维数组。对这个视图的修改会直接反映在原始数组上,从而避免了昂贵的数据复制操作。

理解 shape 和 strides 参数

  • shape: 新视图的形状。对于一个 (ny, nx) 的原始2D数组 A,如果我们要以 (2, 2) 的块进行操作,那么新的视图的形状将是 (ny//2, nx//2, 2, 2)。这表示视图包含 ny//2 行和 nx//2 列,每个“元素”本身又是一个 (2, 2) 的子数组。

  • strides: 定义了在内存中沿着某个维度移动一个单位所需的字节数。

    • 对于原始数组 A,其步幅为 A.strides = (row_stride, col_stride),其中 row_stride 是跳到下一行所需的字节数,col_stride 是跳到下一列所需的字节数。
    • 为了在视图中从一个2x2块的起始位置跳到下一个2x2块的起始位置(即视图的第一和第二个维度),我们需要在原始数组中跳过两行或两列。因此,视图的第一个维度(块行)步幅是 A.strides[0] * 2,第二个维度(块列)步幅是 A.strides[1] * 2。
    • 为了在视图内部的2x2块中移动(即视图的第三和第四个维度),我们需要按照原始数组的行和列步幅来移动。因此,视图的第三个维度(块内行)步幅是 A.strides[0],第四个维度(块内列)步幅是 A.strides[1]。
    • 综合起来,视图的步幅将是 (A.strides[0]*2, A.strides[1]*2, A.strides[0], A.strides[1])。

示例:创建2x2块视图

import numpy as np

# 示例数据:一个10x10的二进制数组
A = np.random.randint(0, 2, (10, 10))
print("原始数组 A:")
print(A)

# 计算视图的形状和步幅
ny, nx = A.shape
block_rows = ny // 2
block_cols = nx // 2

# 创建2x2块的视图
Av = np.lib.stride_tricks.as_strided(A,
                                     shape=(block_rows, block_cols, 2, 2),
                                     strides=(A.strides[0] * 2, A.strides[1] * 2,
                                              A.strides[0], A.strides[1]))
print("\n通过as_strided创建的2x2块视图 Av 的形状:", Av.shape)
print("Av[0,0] (原始数组A的第一个2x2块):\n", Av[0,0])

# 验证视图是否指向原始数据 (修改视图会影响原始数组)
# Av[0,0,0,0] = 99 # 修改视图的第一个元素
# print("\n修改Av[0,0,0,0]后,原始数组A的左上角:\n", A[:2,:2]) # 原始数组A也会随之改变

方法一:基于多维查找表的块替换

当每个2x2块的替换规则依赖于其内部的四个元素时,我们可以构建一个多维查找表(Lookup Table, LUT)。对于由布尔值(0或1)组成的2x2块,共有 2^4 = 16 种可能的状态。查找表将以前四个元素的值作为索引,存储对应的2x2替换块。

剪映
剪映

一款全能易用的桌面端剪辑软件

下载

构建与应用查找表

查找表的形状将是 (2, 2, 2, 2, 2, 2),其中前四个 2 对应输入2x2块的四个元素 (0,0), (0,1), (1,0), (1,1) 的值,后两个 2 对应替换后的2x2块的形状。

# 示例查找表 (lut[val00, val01, val10, val11] = replacement_block)
# lut的形状为 (2,2,2,2, 2,2)
# 前四个2代表2x2块的四个元素 (0,0), (0,1), (1,0), (1,1) 的值
# 后两个2代表替换后的2x2块的形状
lut = np.zeros((2, 2, 2, 2, 2, 2), dtype=A.dtype)

# 填充一些转换规则。例如:
# 1. 原始块 [[0,0],[0,0]] 替换为 [[1,1],[1,1]]
lut[0, 0, 0, 0] = [[1, 1], [1, 1]]
# 2. 原始块 [[0,0],[0,1]] 替换为 [[1,1],[1,0]]
lut[0, 0, 0, 1] = [[1, 1], [1, 0]]
# 3. 原始块 [[0,1],[0,0]] 替换为 [[1,1],[0,1]]
lut[0, 1, 0, 0] = [[1, 1], [0, 1]]
# 4. 原始块 [[1,1],[0,0]] 替换为 [[1,1],[1,1]]
lut[1, 1, 0, 0] = [[1, 1], [1, 1]]
# ... 可以根据需要填充所有16种情况

# 应用查找表进行块替换
# 使用高级索引,将Av中的每个2x2块的四个元素作为lut的索引
Av[:] = lut[Av[..., 0, 0], Av[..., 0, 1], Av[..., 1, 0], Av[..., 1, 1]]

print("\n方法一:使用多维查找表后的数组 A:")
print(A)

方法二:基于一维查找表与索引转换的块替换

为了简化查找表的索引,我们可以将每个2x2块的四个布尔值转换为一个单一的整数索引(0-15)。这可以通过为每个位置分配一个权重并求和来实现。这种方法可以使查找表更紧凑,并可能简化索引逻辑。

索引转换与一维查找表

我们可以定义一个权重矩阵,例如:

[[8, 4],
 [2, 1]]

对于一个2x2块 [[v00, v01], [v10, v11]],其对应的单一索引将是 v00*8 + v01*4 + v10*2 + v11*1。这个索引的范围是0到15。

# 重新初始化数组A进行演示
A = np.random.randint(0, 2, (10, 10))
ny, nx = A.shape
block_rows = ny // 2
block_cols = nx // 2
Av = np.lib.stride_tricks.as_strided(A,
                                     shape=(block_rows, block_cols, 2, 2),
                                     strides=(A.strides[0] * 2, A.strides[1] * 2,
                                              A.strides[0], A.strides[1]))
print("\n重新初始化后的原始数组 A:")
print(A)

# 示例一维查找表 (lut2[index] = replacement_block)
lut2 = np.zeros((16, 2, 2), dtype=A.dtype)

# 填充一些转换规则。例如:
# 索引0 (对应原始块 [[0,0],[0,0]]) 替换为 [[1,1],[1,1]]
lut2[0] = [[1, 1], [1, 1]]
# 索引1 (对应原始块 [[0,0],[0,1]]) 替换为 [[1,1

热门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号