如何从 numpy 中二维数组的第一列创建子数组

Question

使用numpy，怎么可能取数组

np.array([[1,2,3],[4,5,6],[7,8,9]])

并取出数组

[1,4,7] and [[2,3],[5,6],[8,9]]

Answer 1

您可以这样使用索引：

In [9]: a = np.array([[1,2,3],[4,5,6],[7,8,9]])

In [10]: a[:,0]
Out[10]: array([1, 4, 7])

In [11]: a[:,1:]
Out[11]:
array([[2, 3],
       [5, 6],
       [8, 9]])