切片numpy数组的最后2列

Question

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

例如： A = np.array([[1, 2, 3], [4, 5, 6]]) 我想将B作为A的最后2列，即[[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]])