如何使用Numpy动态索引N维？

Question

当我输入N值时，会自动生成N维矩阵。我想根据N的值自动索引N维矩阵。

例如，如果矩阵A的维数为3：

temp = A[a + 1, a + 2, a + 3]

如果是4：

temp = A[a + 1, a + 2, a + 3, a + 4]

因为我的整个代码通过N的值生成了许多具有不同尺寸的矩阵，所以我寻求满足我需求的方法。

感谢。

Answer 1

只需构建一个包含所需索引的元组，并使用该元组索引到temp。