三维数组按第三维度划分np.split示例

下面是一个简单的np.split()函数在三维数组上按第三维度划分的示例：

import numpy as np

arr = 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]]])

sub_arrays = np.split(arr, 3, axis=2)

print(sub_arrays)

输出结果为：

[array([[[ 1],
         [ 4]],
        [[ 7],
         [10]],
        [[13],
         [16]],
        [[19],
         [22]]]), 
 array([[[ 2],
         [ 5]],
        [[ 8],
         [11]],
        [[14],
         [17]],
        [[20],
         [23]]]), 
 array([[[ 3],
         [ 6]],
        [[ 9],
         [12]],
        [[15],
         [18]],
        [[21],
         [24]]])]

上述示例中，我们使用np.array()函数创建了一个三维数组arr，并将其沿着第三个轴（axis=2）的方向分割成了三个子数组。分割后的结果是一个列表sub_arrays，其中包含三个元素，每个元素都是一个分割后的子数组。