np.split示例

下面是一个简单的np.split()函数的示例：

import numpy as np

arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
sub_arrays = np.split(arr, 2, axis=1)

print(sub_arrays)

输出结果为：

[array([[ 1,  2],
       [ 5,  6],
       [ 9, 10]]), 
 array([[ 3,  4],
        [ 7,  8],
        [11, 12]])]

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

相关问题

三维数组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]]])

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

print(sub_arrays)

输出结果为：

[array([[[1, 2],
         [3, 4]],
 
        [[5, 6],
         [7, 8]]]), 
 array([[[ 9, 10],
         [11, 12]],
 
        [[13, 14],
         [15, 16]]])]

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

三维数组按第三维度划分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，其中包含三个元素，每个元素都是一个分割后的子数组。