python提取n列

Question

我有一个格式的numpy数组：

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

我想将元素提取为：

a' = [[1,2],[4,5],[7,8]]

我尝试使用：

a' = a[:][:2]

但它不像我预期的那样工作

Answer 1

>>> [l[:2] for l in a]
[[1, 2], [4, 5], [7, 8]]

Answer 2

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