arrays - 切片numpy数组的最后两列

Question

如何切片 numpy 数组的最后 2 列？

例如： A = np.array([[1, 2, 3], [4, 5, 6]]) 我想将 B 作为 A 的最后两列，即[[2, 3], [5, 6]]

我知道我可以从数组的开头索引它，例如B = A[:, 1:3]. 但是我正在寻找一种通用形式，通过从末尾开始索引来对 A 进行切片，因为在我的情况下 A 的列数会发生变化。

Answer 1

干得好

>>> A = np.array([[1, 2, 3],[4, 5, 6]])
>>> A[:,[-2,-1]]
array([[2, 3],
       [5, 6]])

获取最后 n 行的通用方法可以是

>>> A = np.array([[1, 2, 3,4],[4, 5, 6,7]]) 
>>> A[:,-3:]
array([[2, 3, 4],
       [5, 6, 7]])