切片 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]])

Slicing the last 2 columns of a numpy array