分割数组python

Question

我有一个Python数组中的二维数组，它既可以是普通数组也可以是具有维数（150,5）的numpy数组，我希望将它分为两​​个维数组（150,3）和（150,2）分别。不知何故，我无法做到这一点。

有什么建议吗？

Answer 1

为numpy的阵列可以切片他们是这样的：

a, b = the_array[...,:3], the_array[...,3:] 

，并用列表的列表（这就是我理解的“正常排列”）

a, b = [i[:3] for i in the_array], [i[3:] for i in the_array]