如何将滑动窗口的三维数组展平为二维数组？

import numpy as np

def flatten_3d_array(array, window_size):
    rows, cols, depth = array.shape
    flattened_array = []
    
    for i in range(rows - window_size + 1):
        for j in range(cols - window_size + 1):
            window = array[i:i+window_size, j:j+window_size, :]
            flattened_window = window.flatten()
            flattened_array.append(flattened_window)
    
    return np.array(flattened_array)

# 示例用法
array_3d = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
                     [[10, 11, 12], [13, 14, 15], [16, 17, 18]],
                     [[19, 20, 21], [22, 23, 24], [25, 26, 27]]])

window_size = 2
flattened_array = flatten_3d_array(array_3d, window_size)
print(flattened_array)