NumPy 迭代器对象 numpy.nditer 提供了一种灵活访问一个或者多个数组元素的方式。
迭代器最基本的任务的可以完成对数组元素的访问。
接下来我们使用 arange() 函数创建一个 2X3 数组,并使用 nditer 对它进行迭代。
mport numpy as np a = np.arange(6).reshape(2,3) print ('原始数组是:') print (a) print ('\n') print ('迭代输出元素:') for x in np.nditer(a): print (x, end=", " ) print ('\n')
输出结果为:
原始数组是:
[[0 1 2]
[3 4 5]]
迭代输出元素:0, 1, 2, 3, 4, 5,
数组转置
b=a.T print(b)
输出结果为:
[[0, 3],
[1, 4],
[2, 5]]
如何控制遍历循序
for x in np.nditer(a, order='F'):Fortran order,即是列序优先;
for x in np.nditer(a, order='C'):C order,即是行序优先;
实例如下:
for x in np.nditer(a, order='F'): print(x,end=",")
输出结果为:0,3,1,4,2,5,
for x in np.nditer(a, order='C'): print(x,end=",")
输出结果为:0,1,2,3,4,5,
评论