使用numpy的nditer函数可以遍历numpy数组的每一行。下面是一个示例代码:
import numpy as np
# 创建一个numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 使用nditer函数遍历每一行
for row in np.nditer(arr):
print(row)
输出:
1
2
3
4
5
6
7
8
9
注意:nditer函数是一个迭代器对象,可以通过设置参数来控制遍历的顺序和步长。上面的示例代码中,nditer函数默认按照C风格的顺序(即按行)遍历数组。如果需要按列遍历数组,可以将参数order='F'
传递给nditer函数。
上一篇:遍历numpy数组
下一篇:遍历numpy数组的行