Matlab数组索引转Python切片后输出维度不匹配问题求助

错误原因 #

1. 索引规则不匹配：Matlab中用两个形状为(1,128)的向量分别索引行、列维度时，会自动广播生成(128,128)的网格索引，最终返回同形状的结果；而NumPy的花式索引默认是配对取值，即取[firstAxis[0,i], secondAxis[0,i]]位置的元素，最终只会得到长度为128的一维结果，这是输出形状不符的核心原因。
2. 第三维索引偏移：原Matlab代码第三维索引为kk-1，你的Python代码错误写为kk-2，索引整体偏移1位，直接导致数值不匹配。
3. 多余的矩阵乘法运算：如果N、K为标量参数，直接使用逐元素乘法*即可，你调用的dot为矩阵乘法，在部分数组维度场景下会导致索引计算结果错误。

修正方案 #

调整索引广播逻辑、对齐第三维索引、替换多余的dot运算即可，示例代码如下：

import numpy as np

# 逐元素计算索引，结果转为整型适配NumPy索引要求
firstAxis = np.round(((nn - 1) * N * 2) / K + nN).astype(int)
secondAxis = np.round(((mm - 1) * N * 2) / K + nN).astype(int)

# 将第一个索引转为列向量，触发广播生成128*128的网格索引，同时对齐各维度偏移
xx = wqtran[firstAxis.reshape(-1, 1) - 1, secondAxis - 1, kk - 1]

如果你的wqtran数组是直接从Matlab导出导入到Python的，注意Matlab为列优先存储，Python默认为行优先存储，若存在数值仍不匹配的情况，可在导入wqtran后加一句wqtran = np.ascontiguousarray(wqtran.T)调整存储顺序即可。

内容的提问来源于stack exchange，提问作者material bug