N u m P y NumPy NumPy的 n d i t e r nditer nditer
nditer
是 NumPy 提供的一种多维迭代器
,用于对多维数组进行迭代操作。它可以替代传统的嵌套循环,在处理多维数组时更加方便和高效。
迭代器可以按照不同的顺序
遍历数组的元素,也可以控制迭代器的访问方式
和步
长。在处理多维数组时,特别是需要对数组的每个元素进行操作时,使用 nditer 迭代器可以大大简化代码。
op_flags
:用于指定操作标志,例如 readwrite 表示可以读写数组元素,readonly 表示只能读取数组元素,writeonly 表示只能写入数组元素等。flags
:用于指定迭代器的行为标志,例如 c_index 表示以 C 顺序迭代,f_index 表示以 Fortran 顺序迭代,external_loop 表示以外部循环方式迭代等。op_axes
:用于指定轴的顺序,可以用来控制迭代器遍历轴的顺序。
import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 使用 nditer 迭代器遍历数组
it = np.nditer(arr, flags=['multi_index'])
while not it.finished:
print("当前元素索引:", it.multi_index)
print("当前元素值:", arr[it.multi_index])
it.iternext()
第二个参数如何使用
import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 使用 nditer 迭代器按照 C 顺序(逐行)遍历数组
it = np.nditer(arr, flags=['multi_index', 'c_index'])
while not it.finished:
print("当前元素索引:", it.multi_index)
print("当前元素值:", arr[it.multi_index])
it.iternext()
第三个参数如何使用
import numpy as np
# 创建一个三维数组
arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
# 使用 nditer 迭代器遍历数组,按照指定的轴顺序
it = np.nditer(arr, flags=['multi_index'], op_axes=[(0, 2, 1)])
while not it.finished:
print("当前元素索引:", it.multi_index)
print("当前元素值:", arr[it.multi_index])
it.iternext()