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]])