0

我有一个格式为 numpy 的数组:

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

我想将元素提取为:

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

我尝试使用:

a' = a[:][:2]

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

4

2 回答 2

4
>>> [l[:2] for l in a]
[[1, 2], [4, 5], [7, 8]]
于 2012-04-12T06:11:30.023 回答
4
>>> numpy.array([[1,2,3],[4,5,6],[7,8,9]])[:,:2]
array([[1, 2],
       [4, 5],
       [7, 8]])
于 2012-04-12T06:13:38.873 回答