要遍历NumPy ndarray的所有样本,可以使用for循环遍历数组的每一行。以下是一个示例代码:
import numpy as np
# 创建一个3x4的ndarray
arr = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
# 遍历数组的每一行
for row in arr:
print(row)
输出结果为:
[1 2 3 4]
[5 6 7 8]
[ 9 10 11 12]
如果想要遍历数组的每一个元素,可以使用nditer函数。
import numpy as np
# 创建一个3x4的ndarray
arr = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
# 遍历数组的每一个元素
for element in np.nditer(arr):
print(element)
输出结果为:
1
2
3
4
5
6
7
8
9
10
11
12
使用np.nditer函数可以灵活地控制遍历的方式,例如按照列的顺序遍历。
import numpy as np
# 创建一个3x4的ndarray
arr = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
# 按照列的顺序遍历数组
for column in np.nditer(arr.T):
print(column)
输出结果为:
1
5
9
2
6
10
3
7
11
4
8
12
下一篇:遍历numpy矩阵元素