numpy切片2D数组中的2列

时间:2018-12-12 22:50:37

标签: python arrays numpy

嗨,我正在尝试从2D numpy数组中切出2列,但似乎如果不使用循环就无法获取它。
我想念什么?我正在尝试获取[3 7 11][4 8 12]

import numpy as np

a=np.array([1,2,3,4,5,6,7,8,9,10,11,12])

a=np.reshape(a,(3,4))

print a[:,2] #vgives me [ 3  7 11]

我唯一能做到的方法似乎是循环操作

for i in range(2,4):
    print a[:,i]

如何用纯Numpy切片做到这一点?

1 个答案:

答案 0 :(得分:2)

您要通过此代码获取的元素:

for i in range(2,4):
    print a[:, i]

对应于此:

print a[:, 2:4]