如果我事先不知道Numpy数组的秩，如何索引它的最后一个轴？
下面是我想做的事情：设a是一个秩未知的Numpy数组，我需要最后一个轴的最后一个k元素的切片。
如果a是一维的

b = a[-k:]

如果a是二维的，我想

b = a[:, -k:]

如果a是3D的，我想

b = a[:, :, -k:]

和/或其他信息。
不管a的秩是多少（只要秩至少为1），我都希望它能工作。
在这个例子中，我想要最后的k元素，这是无关紧要的，重点是，当我不知道数组的秩时，我想要指定最后一个轴的索引。