将numpy数组切成两部分

Question

我有一个2d numpy数组 像这样：

 [[ 1 2 3 4],
  [4,5,6,7]..
    ...... ] ]

现在我想把它分成两部分。 让我们说第一个numpy数组有前两个元素。 第二个numpy数组包含其余的元素 像这样的东西

B = [[1 2 3 4],
      [4 5 6 7]]
C = [[ rest of the elements]]

我该怎么做 感谢

Answer 1

本教程的Indexing, Slicing, and Iterating部分对此进行了介绍：

>>> import numpy as np
>>> A = np.array([[1,2,3,4],[4,5,6,7],[7,8,9,10]])
>>> B = A[:2]
>>> C = A[2:]
>>> B
array([[1, 2, 3, 4],
       [4, 5, 6, 7]])
>>> C
array([[ 7,  8,  9, 10]])