нарезать массив кусочков на две части

Question

У меня есть двумерный массив 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

Это рассматривается в Индексировании, нарезке и повторении части урока:

>>> 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]])