12 NumPy 迭代数组

NumPy数组迭代方法

numpy.nditer 是 NumPy 中用于实现数组元素访问的迭代器对象，其设计能够支持对单个或多个数组进行灵活遍历。
该迭代器的核心功能在于实现对数组中各个元素的逐个访问。
随后，我们将通过 arange() 函数生成一个 2 行 3 列的二维数组，并借助 nditer 对其内容进行遍历操作。

实例

导入 numpy 库并定义数组 a，其由 0 至 5 的连续整数构成，并以 2 行 3 列的形式进行重塑。随后输出原始数组的内容，并通过循环遍历数组中的每一个元素，依次打印出每个数值。具体代码如下：

import 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, 
    
    
    
    

上述示例并未采用标