分割数组 Python

Question

我在Python中有一个二维数组，可以是普通数组也可以是numpy数组，大小是(150, 5)。我想把它分成两个数组，一个大小是(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]