如何拆分我的numpy阵列

Question

所以我对numpy很新，我正在尝试一个项目，但遇到了一个我似乎无法解决的错误。

假设我们按以下格式有一个NDarray

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

我将如何把它分成两个部分，使得第一部分是：

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

，第二部分是

[1,0,1] 

我知道解决方案一定很简单，但我似乎无法弄清楚

在此先感谢！

Answer 1

尝试：

a = np.array([[4,5,6,1], 
       [3,5,2,0], 
       [4,7,3,1]]) 

b,c = a[:,:-1], a[:,-1] 

这将使用numpy的的切片，以保持所有行并在最后一个分裂的列。

Answer 2

>>> import numpy as np 
>>> a=np.array([[4,5,6,1],[3,5,2,0],[4,7,3,1]]) 
>>> a 
array([[4, 5, 6, 1], 
     [3, 5, 2, 0], 
     [4, 7, 3, 1]]) 
>>> b=a[:,0:3] 
>>> b 
array([[4, 5, 6], 
     [3, 5, 2], 
     [4, 7, 3]]) 
>>> c=a[:,3] 
>>> c 
array([1, 0, 1]) 
>>> 

这是一种叫做阵列片在python，没有太多numpy的。

有关阵列切片的更多详细信息，请参见Explain Python's slice notation