Python数组组合的一个问题，求高手解答

training_sets = [
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0]
]
我该如何写代码，能够将以上的二维数组，变成以下三维数组
training_sets = [
[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]],
]

import numpy as np

training_sets = np.array( [
    [1,2,3,4,5,6,7,8,9,0],
    [1,2,3,4,5,6,7,8,9,0],
    [1,2,3,4,5,6,7,8,9,0],
    [1,2,3,4,5,6,7,8,9,0],
    [1,2,3,4,5,6,7,8,9,0]
    ])

    training_sets[0:5,0:7]

    training_sets[0:5,7:10]

training_sets = [
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0],
[1,2,3,4,5,6,7,8,9,0]
]

training_sets_final = [];

for training_sets_row in range(0,len(training_sets),1):
    training_sets_temp = [training_sets[training_sets_row][i:i+8] for i in range(0,len(training_sets[0]),8)]
    training_sets_final.append(training_sets_temp)

print(training_sets_final)

[ ([li[0:7],li[7:10]]) for li in training_sets]

https://blog.csdn.net/qq_43287650/article/details/83211898

#定义一个处理函数
def deal(tmp):
return [tmp[:-3],tmp[-3:]] #根据你要分割的位置来设，如果你要分割最后一个，那就是-1

training_sets=list(map(deal,training_sets)) #training_sets 就是你要的结果
print(training_sets)

#运行结果如下：
[[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]],
[[1,2,3,4,5,6,7],[8,9,0]]]