NumPy 教程 之 NumPy 迭代数组 6

NumPy 教程 之 NumPy 迭代数组 6

NumPy 迭代数组

NumPy 迭代器对象 numpy.nditer 提供了一种灵活访问一个或者多个数组元素的方式。

迭代器最基本的任务的可以完成对数组元素的访问。

使用外部循环

nditer 类的构造器拥有 flags 参数，它可以接受下列值：

参数 描述
c_index 可以跟踪 C 顺序的索引
f_index 可以跟踪 Fortran 顺序的索引
multi_index 每次迭代可以跟踪一种索引类型
external_loop 给出的值是具有多个值的一维数组，而不是零维数组
在下面的实例中，迭代器遍历对应于每列，并组合为一维数组。

实例

import numpy as np
a = np.arange(0,60,5)
a = a.reshape(3,4)
print ('原始数组是：')
print (a)
print ('\n')
print ('修改后的数组是：')
for x in np.nditer(a, flags = ['external_loop'], order = 'F'):
print (x, end=", " )

输出结果为：

原始数组是：
[[ 0 5 10 15]
[20 25 30 35]
[40 45 50 55]]

修改后的数组是：
[ 0 20 40], [ 5 25 45], [10 30 50], [15 35 55],