如何 select 矩阵中的行或列？

Question

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

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

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

不过我试过a[:,[0,2]]，但是显示

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

好像不对，又试了一个[:][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]])

How to select row or column from a matrix?