python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > nditer numpy.ndarray多维数组迭代

nditer—numpy.ndarray 多维数组的迭代操作

作者:Inside_Zhang

这篇文章主要介绍了nditer—numpy.ndarray 多维数组的迭代操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

1. Single array iteration

>>> a = np.arange(6).reshape(2,3)
>>> for x in np.nditer(a):
...     print x,
...
0 1 2 3 4 5

也即默认是行序优先(row-major order,或者说是 C-order),这样迭代遍历的目的在于,实现和内存分布格局的一致性,以提升访问的便捷性;

>>> for x in np.nditer(a.T):
...     print x,
...
0 1 2 3 4 5
>>> for x in np.nditer(a.T.copy(order='C')):
...     print x,
...
0 3 1 4 2 5

也即对 a 和 a.T 的遍历执行的是同意顺序,也即是它们在内存中的实际存储顺序。

2. 控制遍历顺序

for x in np.nditer(a, order='F'):Fortran order,也即是列序优先;
for x in np.nditer(a.T, order='C'):C order,也即是行序优先;

3. 修改数组中元素的值

默认情况下,nditer将视待迭代遍历的数组为只读对象(read-only),为了在遍历数组的同时,实现对数组元素值得修改,必须指定 read-write 或者 write-only的模式。

>>> a
array([[0, 1, 2],
       [3, 4, 5]])
>>> for x in np.nditer(a, op_flags=['readwrite']):
...     x[...] = 2 * x
...
>>> a
array([[ 0,  2,  4],
       [ 6,  8, 10]])

4. 使用外部循环

将一维的最内层的循环转移到外部循环迭代器,使得 numpy 的矢量化操作在处理更大规模数据时变得更有效率。

>>> a = np.arange(6).reshape(2,3)
>>> for x in np.nditer(a, flags=['external_loop']):
...     print x,
...
[0 1 2 3 4 5]
>>>
>>> for x in np.nditer(a, flags=['external_loop'], order='F'):
...     print x,
...
[0 3] [1 4] [2 5]

5. 追踪单个索引或多重索引(multi-index)

>>> a = np.arange(6).reshape(2,3)
>>> a
array([[0, 1, 2],
       [3, 4, 5]])
>>> it = np.nditer(a, flags=['f_index'])
>>> while not it.finished:
...     print "%d <%d>" % (it[0], it.index),
...     it.iternext()
...
0 <0> 1 <2> 2 <4> 3 <1> 4 <3> 5 <5>
            # 索引的编号,以列序优先
>>> it = np.nditer(a, flags=['multi_index'])
>>> while not it.finished:
...     print "%d <%s>" % (it[0], it.multi_index),
...     it.iternext()
...
0 <(0, 0)> 1 <(0, 1)> 2 <(0, 2)> 3 <(1, 0)> 4 <(1, 1)> 5 <(1, 2)>

补充:详解 Numpy.ndarray

向量、矩阵 & 多维数组是数值计算中必不可少的工具;通过对数组数据进行批量处理,避免了对数组元素显式地进行循环操作,这样做的结果是可以得到简洁、更易维护的代码,并且可以使用更底层的库来实现数组操作。因此,向量化计算相比按顺序逐元素进行计算要快得多。

在 Python科学计算环境中,Numpy 库提供了用于处理数组的高效数据结构,且Numpy的核心是使用C语言实现的,提供了很多处理和处理数组的函数。

NumPy支持比Python更多种类的数字类型,有5种基本数字类型:

布尔值(bool)

整数(int)

无符号整数(uint)

浮点(float)

复数(complex)

Numpy库的核心是表示 同质的多维数据 —— 每个元素占用相同大小的内存块, 并且所有块都以完全相同的方式解释。 如何解释数组中的每个元素由单独的数据类型对象指定, 其中一个对象与每个数组相关联。除了基本类型(整数,浮点数 等 )之外, 数据类型对象还可以表示数据结构。

1、创建 Numpy 数组

NumPy提供了一个N维数组类型,即ndarray, 它描述了相同类型的“项目”集合。可以使用例如N个整数来索引项目。从数组中提取的项( 例如 ,通过索引)由Python对象表示, 其类型是在NumPy中构建的数组标量类型之一。 数组标量允许容易地操纵更复杂的数据排列。

ndarray 与 array 的区别

np.array 只是一个便捷的函数,用来创建一个ndarray,它本身不是一个类。

ndarray 数组,是用 np.ndarray类的对象 表示n维数组对象

所以ndarray是一个类对象,而array是一个方法。

创建数组有5种常规机制:

从其他Python结构(例如,列表,元组)转换

numpy原生数组的创建(例如,arange、ones、zeros等)

从磁盘读取数组,无论是标准格式还是自定义格式

通过使用字符串或缓冲区从原始字节创建数组

使用特殊库函数(例如,random)

1、np.array

一个 ndarray是具有相同类型和大小的项目的(通常是固定大小的)多维容器。 尺寸和数组中的项目的数量是由它的shape定义, 它是由N个非负整数组成的tuple(元组),用于指定每个维度的大小。 数组中项目的类型由单独的data-type object (dtype)指定, 其中一个与每个ndarray相关联。

与Python中的其他容器对象一样,可以通过对数组进行索引或切片(例如,使用N个整数)以及通过ndarray的方法和属性来访问和修改ndarray的内容。

不同的是,ndarrays可以共享相同的数据, 因此在一个ndarray中进行的更改可能在另一个中可见。 也就是说,ndarray可以是另一个ndarray 的 “view” ,它所指的数据由 “base” ndarray处理。 ndarrays也可以是Python拥有的内存strings或实现 buffer 或数组接口的对象的视图。

通过 np.array() & np.ndarray() 创建

# Create an array.
np.array(object, dtype=None, copy=True, order='K', subok=False, ndmin=0)
np.ndarray(shape, dtype=float, buffer=None, offset=0, strides=None, order=None)

一个 ndarray是具有相同类型和大小的项目的多维容器。

尺寸和数组中的项目的数量是由它的shape定义, 它是由N个非负整数组成的tuple(元组),用于指定每个维度的大小。

不同的是,ndarrays可以共享相同的数据, 因此在一个ndarray中进行的更改可能在另一个中可见。 也就是说,ndarray可以是另一个ndarray 的 “view” ,它所指的数据由 “base” ndarray处理。 ndarrays也可以是Python拥有的内存strings或实现 buffer 或数组接口的对象的视图。

Examples:

>>> np.array([1, 2, 3])
array([1, 2, 3])
>>> np.array([1, 2, 3.0])
array([ 1., 2., 3.])
>>> np.array([[1, 2], [3, 4]])
array([[1, 2],
[3, 4]])
>>> np.array([1, 2, 3], ndmin=2)
array([[1, 2, 3]])
>>> np.array([1, 2, 3], dtype=complex)
array([ 1.+0.j, 2.+0.j, 3.+0.j])
>>> x = np.array([(1,2),(3,4)],dtype=[('a','<i4'),('b','<i4')])
>>> x['a']
array([1, 3])
>>> np.array(np.mat('1 2; 3 4'))
array([[1, 2],
[3, 4]])
>>> np.array(np.mat('1 2; 3 4'), subok=True)
matrix([[1, 2],
[3, 4]])
>>> np.ndarray(shape=(2,2), dtype=float, order='F')
array([[ -1.13698227e+002, 4.25087011e-303],
[ 2.88528414e-306, 3.27025015e-309]])  #random
>>> np.ndarray((2,), buffer=np.array([1,2,3]),
offset=np.int_().itemsize,
dtype=int)         # offset = 1 * itemsize, i.e. skip first element
array([2, 3])

2、基本属性

数组属性反映了数组本身固有的信息。通常,通过其属性访问数组允许您获取并有时设置数组的内部属性,而无需创建新数组。公开的属性是数组的核心部分,只有一些属性可以有意义地重置而无需创建新数组。有关每个属性的信息如下。

内存布局

以下属性包含有关数组内存布局的信息:

   方法        描   述
| ndarray.flags  | 有关数组内存布局的信息。     
| ndarray.shape  | 数组维度的元组。       
| ndarray.strides  | 遍历数组时每个维度中的字节元组。   
| ndarray.ndim  | 数组维数。        
| ndarray.data  | Python缓冲区对象指向数组的数据的开头。  
| ndarray.size  | 数组中的元素数。       
| ndarray.itemsize | 一个数组元素的长度,以字节为单位。   
| ndarray.nbytes  | 数组元素消耗的总字节数。     
| ndarray.base  | 如果内存来自其他对象,则为基础对象。

数据类型

可以在dtype属性中找到与该数组关联的数据类型对象 :

   方法   |  描   述
| ndarray.dtype  | 数组元素的数据类型。     

其他属性

   方法   |  描   述
| ndarray.T   | 转置数组。        
| ndarray.real  | 数组的真实部分。       
| ndarray.imag  | 数组的虚部。        
| ndarray.flat  | 数组上的一维迭代器。      
| ndarray.ctypes  | 一个简化数组与ctypes模块交互的对象。  

3、Numpy 原生数组 创建 ndarray

       方法            |          描   述 
| eye(N[, M, k, dtype, order])       | 返回一个二维数组,对角线上有一个,其他地方为零
| identity(n[, dtype])         | 返回标识数组。 
| ones(shape[, dtype, order])       | 返回给定形状和类型的新数组,并填充为1
| ones_like(a[, dtype, order, subok, shape])    | 返回形状与类型与给定数组相同的数组。
| zeros(shape[, dtype, order])       | 返回给定形状和类型的新数组,并用零填充。
| zeros_like(a[, dtype, order, subok, shape])   | 返回形状与类型与给定数组相同的零数组。 
| full(shape, fill_value[, dtype, order])    | 返回给定形状和类型的新数组,并用fill_value填充
| full_like(a, fill_value[, dtype, order, …])   | 返回形状和类型与给定数组相同的完整数组
| empty(shape[, dtype, order])       | 返回给定形状和类型的新数组,而无需初始化条目
| empty_like(prototype[, dtype, order, subok, …])  | 返回形状和类型与给定数组相同的新数组

zeros_like()、ones_like()、empty_like() 等带 _like() 的函数创建与参数数组的形状及类型相同的数组。

frombuffer()、fromstring()、fromfile() 等函数可以从字节序列或文件创建数组

4、np.arange

|      方法                       |   描   述 
|  arange([start,] stop[, step,][, dtype])     | 返回给定间隔内的均匀间隔的值。
|  linspace(start, stop[, num, endpoint, …])     | 返回指定间隔内的等间隔数字。
|  logspace(start, stop[, num, endpoint, base, …])  |  返回数以对数刻度均匀分布。  
|  geomspace(start, stop[, num, endpoint, …])    | 返回数字以对数刻度(几何级数)均匀分布。       
|  meshgrid(*xi, **kwargs)         | 从坐标向量返回坐标矩阵。 
| mgridnd_grid            | 实例,它返回一个密集的多维 “meshgrid”
| ogridnd_grid            | 实例,它返回一个开放的多维 “meshgrid”
        

2、从现有的数据创建

      方法            描   述
| array(object[, dtype, copy, order, subok, ndmin]) | 创建一个数组
| asarray(a[, dtype, order])       | 将输入转换为数组
| asanyarray(a[, dtype, order])      | 将输入转换为ndarray,但通过ndarray子类
| ascontiguousarray(a[, dtype])      | 返回内存中的连续数组(ndim > = 1)(C顺序)
| asmatrix(data[, dtype])        | 将输入解释为矩阵
| copy(a[, order])         | 返回给定对象的数组副本
| frombuffer(buffer[, dtype, count, offset])   | 将缓冲区解释为一维数组
| fromfile(file[, dtype, count, sep, offset])   | 根据文本或二进制文件中的数据构造一个数组
| fromfunction(function, shape, **kwargs)    | 通过在每个坐标上执行一个函数来构造一个数组
| fromiter(iterable, dtype[, count])     | 从可迭代对象创建一个新的一维数组
| fromstring(string[, dtype, count, sep])    | 从字符串中的文本数据初始化的新一维数组
| loadtxt(fname[, dtype, comments, delimiter, …])  | 从文本文件加载数据

3、创建矩阵

    方法                                 |     描 述  
| mat(data[, dtype])           | 将输入解释为矩阵
| bmat(obj[, ldict, gdict])         | 从字符串,嵌套序列或数组构建矩阵对象
|  tril(m[, k])           |  数组的下三角。                               
|  triu(m[, k])           |  数组的上三角。                               
|  vander(x[, N, increasing])        | 生成范德蒙矩阵        
|  diag(v[, k])           |  提取对角线或构造对角线数组。                 
|  diagflat(v[, k])          |  使用展平的输入作为对角线创建二维数组。       
|  tri(N[, M, k, dtype])         |  在给定对角线处及以下且在其他位置为零的数组。 

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。

您可能感兴趣的文章:
阅读全文