分割numpy数组的最后2列

Question

如何分割numpy数组的最后2列

例如：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]])