如何在MATLAB中转换此嵌套为循环的Python?

时间:2018-11-03 08:13:44

标签: python matlab loops for-loop matrix

我需要转换以下嵌套在MATLAB中的for循环的Python,但是它不起作用。

for ii in where(obs != 0)[0]:
    ob = obs[ii]
    ncount = int(floor(top / ob))
    for jj in range(ncount):
        l_ind = max(int(floor((ob * (jj + 1) - width / 2.) / top * (G - 1))), 0)
        h_ind = min(int(ceil((ob * (jj + 1) + width / 2.) / top * (G - 1))), G - 1)
        ret[ii, l_ind : h_ind + 1] = tukey(h_ind - l_ind + 1, 0.5)

我个人不需要“ where(obs!= 0)[0]”部分,因为我的obs向量在该维度上没有零。到目前为止,我尝试了以下方法:

for ii = 1:length(obs)
    ob = obs(ii);
    ncount = floor(top / ob);
    for jj = 1:ncount
        l_ind = max(floor((ob * (jj + 1) - width / 2) / top * (G - 1)), 0);
        h_ind = min(ceil((ob * (jj + 1) + width / 2) / top * (G - 1)), G - 1);
        ret(ii , l_ind : h_ind) = tukeywin(h_ind - l_ind + 1)';
    end
end

在Matlab中,tukey称为tukeywin,我不需要额外的0.5参数,因为它是默认值。如您所见,我还必须转置tukeywin来匹配大小,否则会给我另一个错误,并从h_ind中删除+ 1,这可能是因为在Python中,索引从0开始,在Matlab中,索引从开始1.如果保持+ 1,则必须在tukeywin函数中放入+ 2,以匹配方程式两边的大小。

但是在那之后,它仍然不起作用,我在生成ret矩阵的最后一行上遇到了问题。 ret矩阵用零初始化,大小为1972 x1025。obs向量的大小为1972 x1。G为1025,宽度为300,顶部为22050。obs向量包含范围在100-600之间的数字,例如您应该能够重新创建此嵌套循环。

它给我错误“ 位置2的索引无效。数组索引必须为正整数或逻辑值”。我假设这是指ret(ii,l_ind:h_ind)部分,ii很好,它将一直持续到1972年,但是l_ind:h_ind给了我我认为的错误。所以我不确定如何在Matlab中构造它。

1 个答案:

答案 0 :(得分:0)

在没有完全测试您的代码的情况下,我认为您的问题与基于0对1的数组索引有关。您将l_indh_ind限制为0和G-1。对于Matlab,您需要1和G。

我不太确定您在floor / ceil语句中正在做什么以确定索引,但是我怀疑逻辑将需要进行一些调整。最简单的解决方案可能只是对l_indh_ind行进行+1。否则,您需要使用基于1的索引对其进行重做...