在Python中从3D到2D

时间:2013-06-25 15:46:30

标签: python numpy scipy ipython

我想要这段代码:

chains3D = [[] for k in range(colors)]
        def addr(x,y,z): return x + (nx) * (y + (ny) * (z))
        for x in range(nx):
            for y in range(ny):
                for z in range(nz):
                    if (image[x,y,z] == background):
                        chains3D[1].append(addr(x,y,z))
                    else:
                        chains3D[0].append(addr(x,y,z))

就像这样:

chains2D = [[] for k in range(colors)]
        def addr(x,y): return x + (nx) * y 
        for x in range(nx):
            for y in range(ny):
                    if (image[x,y,0] == background):
                        chains2D[1].append(addr(x,y))
                    else:
                        chains2D[0].append(addr(x,y))

好的,我已经解决了代码问题,但现在我遇到了这个错误:

IndexError                                Traceback (most recent call last)
<ipython-input-5-6a7d44bd72b7> in <module>()
213                         objectBoundaryChain = larBoundaryChain(partial_3,chains2D[1])
214                         b2cells = csrChainToCellList(objectBoundaryChain)
--> 215                         sup_cell_boundary = MKPOLS((V,[FV[f] for f in b2cells]))
216 
217                         # remove the (local) boundary (shared with the piece boundary)     from the quotient cell

/Users/Fabio/larpy/lar.pyc in MKPOLS(model)
101     """
102     V, FV = model
--> 103     pols = [MKPOL([[V[v] for v in f],[range(1,len(f)+1)], None]) for f in FV]
104     return pols
105 

IndexError: list index out of range

我不知道为什么f等于b2cells的最后一个数,而不是第一个,但也许这不是产生这个错误的真正问题

1 个答案:

答案 0 :(得分:2)

目前还不完全清楚你要做什么和什么不行,但看起来很奇怪的是你改变了:

return x + (nx) * (y + (ny) * (z))

成:

return x + (nx) * (y + (ny))

不应该是:

return x + (nx) * y

代替?考虑到nz没有出现在您的3D版本中,ny不应出现在2D版本中。