要遍历一个numpy数组的行,可以使用numpy.nditer
函数来实现。下面是一个示例代码:
import numpy as np
# 创建一个3x3的numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 使用nditer函数遍历数组的行
for row in np.nditer(arr, flags=['external_loop'], order='C'):
print(row)
输出结果如下:
[1 2 3]
[4 5 6]
[7 8 9]
在上面的代码中,我们使用np.nditer
函数来遍历数组的行。flags=['external_loop']
参数表示我们要遍历数组的外部循环(即行),order='C'
表示按照C顺序(逐行)遍历数组。
注意:numpy.nditer
函数的返回结果是一个可迭代对象,我们可以使用for
循环逐行遍历数组。每一次循环,row
变量都表示数组的一行。
上一篇:遍历numpy数组的每一行
下一篇:遍历numpy数组的最后维度