在python中展平/重塑3D数组

import numpy as np

# 创建一个3D数组
arr_3d = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])

# 展平3D数组
flatten_arr = arr_3d.flatten()
# 或者使用 ravel() 函数
# flatten_arr = arr_3d.ravel()

print(flatten_arr)

[ 1  2  3  4  5  6  7  8  9 10 11 12]

import numpy as np

# 创建一个3D数组
arr_3d = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])

# 重塑3D数组为2D数组
reshape_arr = arr_3d.reshape((2, 6))

print(reshape_arr)

[[ 1  2  3  4  5  6]
 [ 7  8  9 10 11 12]]