我在 Python 中有一个二维数组,可以是普通数组,也可以是维度为 (150, 5) 的 numpy 数组,我希望将其分别拆分为维度为 (150, 3) 和 (150, 2) 的两个数组。不知怎的,我一直没能做到。
有什么建议吗?
最佳答案
对于 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]
关于python - python 分割数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/7426583/