2012-03-19 139 views
2

我有一个2D numpy的阵列 事情是这样的:切片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]] 

东西我如何做到这一点 由于

回答

7

这是覆盖在该教程的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]])