從二維 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]]