将MATLAB转换为Python:太多索引返回错误

时间:2015-06-05 15:19:22

标签: python arrays matlab numpy indexing

我正在尝试将我在MATLAB中使用的脚本转换为Python,以提高整体算法的速度和效率。在MATLAB中,代码如下:

for iter = 1:T
costi = costo;
for i = 1:length(index)
    for j = i+1:length(index)
        if index(j) == index(i)
            continue;
        end
        indexdn = indexd;
        indadd = (index(j) - index(i));
        indexdn(:,j) = indexdn(:,j) + indadd;
        ##line 11
        indexdn(j,:) = -indexdn(:,j)';            
        indexdn(j,j) = 0;
        indi = abs(indexdn);
        indi = ~indi;
        costnb = costmata.*indi;
        costn = 0.5*(sum(sum(costnb)));
        if costn < costi
            costi = costn;
            index(j) = index(i);
            indexd = indexdn;
        end
    end
end
if costi < costo
    costo = costi;
else 
    break
end
iter
end

我已完成大部分翻译:

for j in range(0,T):
cost2 = cost1
for x in xrange(len(index)):
    for y in xrange(x+1,len(index)):
        if index[y] == index[x]:
            continue
        indexdn = indexd
        indadd= index[y]-index[x]
        print indadd
        indexdn[:,y]=indexdn[:,y]+ indadd
        index[y,:]=-indexdn[:,y] ##line 11, return error
        indexdn[y,y]=0
        indi= np.abs(indexdn)
        indi= ~indi
        costnb = costmata*indi
        costn = .5(np.sum(costnb))
        if (costn < cost2):
            costi=costn;
            index[y] = index[x]
            indexd= indexdn
if cost2<cost1:
    cost1=cost2
else:
    break

然而,在第11行,我返回了“索引错误:索引太多”的错误。是什么导致Python在这条线上被绊倒?如何编写我的Python代码以便我不返回此错误? index 数组是一个预定义的numpy数组,长度为16,随机整数为0-5, indexd 数组是一个16x16数组,随机整数为-5到5, indexdn ,indadd 正在这次迭代中创建。

1 个答案:

答案 0 :(得分:3)

看起来index是一维数组吗? (第5行和第8行有index[y]index[x],并且说长度为16)

但是,在第11行,您正尝试访问其第二维:index[y,:]。也许这应该是indexdn[y,:] =-indexdn[:,y]