我试图用一个由我希望保留的索引组成的矩阵来索引3D矩阵的最后一个维度
我有一个带形状的推力值矩阵:
(3, 3, 5)
我想根据一些标准过滤最后一个索引,以便它从大小5减少到大小1。我已经在最后一个维度中找到了符合我标准的索引:
[[0 0 1]
[0 0 1]
[1 4 4]]
我想要实现的是:对于第一行和第一列,我想要最后一个维度的第0个索引。对于第一行和第三列,我需要最后一个维度的第一个索引。就保留最终矩阵的索引而言,最终矩阵将变成(3, 3)
2D矩阵,如下所示:
[[0,0,0], [0,1,0], [0,2,1];
[1,0,0], [1,1,0], [1,2,1];
[2,0,1], [2,1,4], [2,2,4]]
我很有信心numpy能做到这一点,但我不知道具体是如何做到的。我不希望构建一个嵌套for循环的构造
我已经试过:
minValidPower = totalPower[:, :, tuple(indexMatrix)]
但是这会产生一个(3, 3, 3, 3)
矩阵,所以我不完全确定应该如何处理这个问题
可以为前两个维度构建相应的索引数组。这些基本上是:
可以使用meshgrid函数构造这些。在示例中,我将它们存储为m1和m2:
结果的形状与索引数组m1、m2和m3的形状匹配
以
a
作为输入数组,以idx
作为索引数组-或者,使用开放网格-
相关问题 更多 >
编程相关推荐