在numpy中迭代任意维度

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 迭代二维数组
for row in arr:
    for element in row:
        print(element)

# 迭代三维数组
arr_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
for matrix in arr_3d:
    for row in matrix:
        for element in row:
            print(element)

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 迭代二维数组
for element in np.nditer(arr):
    print(element)

# 迭代三维数组
arr_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
for element in np.nditer(arr_3d):
    print(element)

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 迭代二维数组
for element in arr.flat:
    print(element)

# 迭代三维数组
arr_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
for element in arr_3d.flat:
    print(element)