如何使用Numpy动态索引N维数组?

4
当我输入一个值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的值生成许多不同维度的矩阵,所以我正在寻找满足我的需求的方法。

谢谢。


你是指方阵吗? - asheeshr
1个回答

3

只需构建一个包含所需索引的元组,并使用该元组来索引temp


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接