如何提取Numpy矩阵的前n列?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (193)

我有这样一个数组:

  array([[-0.57098887, -0.4274751 , -0.38459931, -0.58593526],
         [-0.22279713, -0.51723555,  0.82462029,  0.05319973],
         [ 0.67492385, -0.69294472, -0.2531966 ,  0.01403201],
         [ 0.41086611,  0.26374238,  0.32859738, -0.80848795]])

现在我想提取以下内容:

   [-0.57098887, -0.4274751]
   [-0.22279713, -0.51723555]
   [ 0.67492385, -0.69294472]
   [ 0.41086611,  0.26374238]

所以基本上只有前两列。

提问于
用户回答回答于

如果a是你的数组:

In [11]: a[:,:2]
Out[11]: 
array([[-0.57098887, -0.4274751 ],
       [-0.22279713, -0.51723555],
       [ 0.67492385, -0.69294472],
       [ 0.41086611,  0.26374238]])
用户回答回答于

我知道这是个很老的问题:

A = [[1,2,3],[4,5,6],[7,8,9],[10,11,12]]

假设您想提取-前2行和前3列

A_NEW = A[0:2,0:3]
A_NEW = [[1,2,3],[4,5,6]]

理解语法

A_NEW = A[Start_index : stop_index, start_index : stop_index)]

如果想要第2行和第2和第3栏

A_NEW = A[1:2,1:3]

扫码关注云+社区