如果我事先不知道它的等级，我如何索引Numpy数组的最后一个轴？
这就是我想要做的事情：让a成为未知等级的Numpy数组。我想要最后一个轴的最后k个元素的切片。
如果a是1D，我想要
b = a[-k:]
如果a是2D，我想要
b = a[:, -k:]
如果a是3D，我想要
b = a[:, :, -k:]
等等。
我希望无论a的排名如何（只要排名至少为1），我都希望这项工作能够实现。
我想要示例中的最后k元素的事实当然是无关紧要的，关键是我想要指定最后一个轴的索引，当我不知道等级的时候提前一个数组。