python将二维数组沿第一维度拆分为三维数组

可以使用 `numpy.split()` 函数将一个二维数组沿着第一维度拆分为多个子数组，然后使用 `numpy.stack()` 函数将这些子数组组合成一个三维数组。

假设我们有一个二维数组 `arr`，形状为 `(m, n)`，我们想将其沿着第一维度拆分为 `l` 个子数组，每个子数组形状为 `(m/l, n)`，然后将这些子数组沿着第一维度组合成一个三维数组，形状为 `(l, m/l, n)`。

下面是一个实现的示例代码：

import numpy as np

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

# 设置拆分数量
l = 2

# 使用 split 函数拆分为多个子数组
sub_arrs = np.split(arr, l)

# 使用 stack 函数组合为三维数组
new_arr = np.stack(sub_arrs)

# 打印新数组
print(new_arr)

在上面的示例中，我们将一个形状为 `(4, 3)` 的二维数组 `arr` 沿着第一维度拆分为 `l=2` 个子数组，每个子数组形状为 `(2, 3)`，然后将这些子数组沿着第一维度组合成一个三维数组，打印输出如下：

array([[[ 1,  2,  3],
        [ 4,  5,  6]],

       [[ 7,  8,  9],
        [10, 11, 12]]])

可以看到，原来的二维数组被拆分为了两个形状为 `(2, 3)` 的子数组，并按照新形状被重组为了一个三维数组。