嗨,我正在尝试从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切片做到这一点?
答案 0 :(得分:2)
您要通过此代码获取的元素:
for i in range(2,4):
print a[:, i]
对应于此:
print a[:, 2:4]