从嵌套列表中删除项目Python

Question

我试图从Python中的嵌套列表中删除项目。我有一个嵌套列表如下：

families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]] 

我想删除在每个子列表是coorespond到主列表中的子表的索引位置的条目。所以，例如，我需要从第一个子列表中删除0，从第二个子列表中删除1，等等。我试图使用列表解析来做到这一点。这是我曾尝试：

familiesNew = [ [ families[i][j] for j in families[i] if i !=j ] for i in range(len(families)) ] 

本工程为range(len(families))最多3个，但除此之外，我得到IndexError: list index out of range。我不知道为什么。有人可以给我一个如何做到这一点的想法。最好是单行（列表理解）。

谢谢。

Answer 1

你几乎说得对。只需更换families[i][j]与j和它的作品：

>>> [ [ j for j in families[i] if i !=j ] for i in range(len(families)) ] 
[[1, 2], [0, 2, 3], [0, 1, 3, 4], [1, 2, 4, 5], [2, 3, 5, 6]] 

它可以写成使用enumerate功能有点清洁：

>>> [[f for f in family if f != i] for i, family in enumerate(families)] 
[[1, 2], [0, 2, 3], [0, 1, 3, 4], [1, 2, 4, 5], [2, 3, 5, 6]] 

甚至使用remove，如果你不介意改变原来的列表：

>>> for i, family in enumerate(families): family.remove(i) 

Answer 2

这是否你想要做什么？

familiesNew=[ filter(lambda x:x!=i,j) for i,j in enumerate(families) ] 

编辑

还要注意，你失败的原因是因为在外部列表（[1, 2, 3, 4, 5]）的第三个元素，你正在试图获得第五元素在你的for循环（for j in families[i] = = for j in [1,2,3,4,5]），但家庭[I]有长度为5，这意味着最大的指数是4，很抱歉，如果这个解释是有点不清楚......也许下面将帮助它清除掉一点：

families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]] 

def f(i,j): 
    print i,j,families[i] 
    return families[i][j] 
#THIS DOES NOT WORK -- but it will tell you where it failed. 
familiesNew = [ [ f(i,j) for j in families[i] if i !=j ] for i in range(len(families)) ] 

Answer 3

编辑的问题，删除我的答案是解决了错误的问题。此外，增加了额外的答案被@Ashwini：

为了比较：

root# python -m timeit 'families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]]' '[x.remove(ind) for ind,x in enumerate(families) ]' 
100000 loops, best of 3: 3.42 usec per loop  

root# python -m timeit -s 'families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]]' '[[f for f in family if f != i] for i, family in enumerate(families)]' 
100000 loops, best of 3: 4.87 usec per loop 

root# python -m timeit -s 'families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]]' '[ filter(lambda x:x!=i,j) for i,j in enumerate(families) ]' 
100000 loops, best of 3: 7.99 usec per loop 

这些都是微秒，所以我想你想干什么就干什么是好的，除非你打算做这样的次。

Answer 4

如果要修改原来的列表，然后试试这个：

>>>[x.remove(ind) for ind,x in enumerate(families) ] 
>>>families 
[[1, 2], [0, 2, 3], [0, 1, 3, 4], [1, 2, 4, 5], [2, 3, 5, 6]]