从二维 numpy 数组中提取列

Question

我有二维 numpy 数组：

import numpy as np

a = np.array([[1,2,3,4],
              [4,5,6,7]])

如何提取以下数组？

result = array([[3,4],
               [6,7]]

Answer 1

这称为slicing ，您可以使用<array>[<row indexes>,<column indexes>] 。

例子 -

import numpy as np

a = np.array([[1,2,3,4],
              [4,5,6,7]])

print(a[:,2:])
>>> [[3 4]
     [6 7]]