python - 如何从矩阵中选择行或列？

Question

这里我有一个矩阵a=np.array([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15]])

我想选择所有行，但我想选择的列是从第一行到第三行。

它应该是[[1,2,3],[6,7,8],[11,12,13]]

但是，我曾经尝试过a[:,[0,2]]，但它显示

 array([[ 1,  3],
       [ 6,  8],
       [11, 13]]) 

好像不正确，于是又试了一个a[:][0:2]，结果还是错误。

所以我想问一下有没有什么功能或者方法可以解决这个问题？

Answer 1

听起来你正在寻找a[:, 0:3]：

In [4]: a[:, 0:3]
Out[4]:
array([[ 1,  2,  3],
       [ 6,  7,  8],
       [11, 12, 13]])

Answer 2

我认为需要索引0:3：

print (a[:,0:3])
[[ 1  2  3]
 [ 6  7  8]
 [11 12 13]]

Answer 3

尝试以下

a=np.array([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15]])
a = a[:,0:3] 
print(a)
#Output
#array([[ 1,  2,  3],
#   [ 6,  7,  8],
#   [11, 12, 13]])