python - 如何使用 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.