NumPy 中的多维迭代器.pdfVIP

NumPy 中的多维迭代器 《代码之美》 第19章 江涌 NumPy是Python的一个可选安装包，提供了一个功能强 大的N维数组对象。NumPy提供了多种数组的数学化操作 与结构化操作，使得Python能够很好地开发一些关键的并 且要求运行速度很快的工程代码和科学代码。 NumPy中通过切片 (slicing)的概念来实现快速结构化操作。 记法为[start:stop:stride]，例如im2=im[8:2:-1,9:1:-3]。 按照slicing方式选取的新影像将与原始影像共享数据，不会 生成一个副本，减少计算机资源的消耗。 关键挑战 经常需要遍历数组中的元素，在遍历中进行所需要的操作。 简单想法：用单层for循环处理一维，双层for循环处理两 维……但当维数N是一个任意整数，怎么办？ 递归：递归条件(recursive case)，基线条件(base case) Copy_ND(a,b,N) //将N维数组b复制到N维数组a 递归实现：if (N==0) copy memory from b to a return for i=0 to size of first dimension of a and b ptr_b=b[i] Copy_ND(ptr_a,ptr_b,N-1) a[i]=ptr_a 递归算法在每次迭代中进行函数调用，容易产生速度很慢的 代码；许多算法需要保存中间值用于后续的递归调用(求最大 值) ，这些值将被作为递归调用的参数传递，很难提供用于递 归解决方案的简化工具。 因此，NumPy使用迭代来完成。迭代器(Iterator)是一种简化这 些算法的抽象，包含了单个循环内遍历数组中所有元素的思 想。迭代器的两个基本方法： hasnext 是否还有下一个元素; next 返回下一个元素。 for x in iterobj: process(x) 数组的内存模型 邻接型数组：在内存中连续存放。 一个二维4*5数组 p=[[1,2,3,4,5], 1 2 3 4 5 [6,7,8,9,10], [11,12,13,14,15], 6 7 8 9 10 [16,17,18,19,20]] 11 12 13 14 15 from numpy import * 16 17 18 19 20 pp=array(p) p1=pp[1:3,1:4] 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 跨度 数组中某一维的跨度(stride) ： 沿着这一维，或者说数轴，从数组中的一个元素移 动到下一个元素，需要跳过多少字节。 （跨度可以是负数） pp: 第一维跨度4*5,第二维跨度4 ； p1 ：第一维跨度4*5 ，第二维跨度4. 迭代器设计 迭代器循环伪码： set up iterator (including pointing the current value to the first value in the array) while iterator not done: process the current value point the current value to the next value 设计分为三部分： 1. Moving to the next value