python - 在 Python 中，如何从列表列表中删除某个元素？

Question

假设我有一个列表列表：

x = [[1,0,0,3],[1,1,1,1],[5,2,0,0],[4,3,0,1],[0,0,0,0]

如何创建一个仅包含每个列表的非零项的新列表，这样

y = [[1,3],[1,1,1,1],[5,2],[4,3,1],[]]

Answer 1

In [23]: x = [[1,0,0,3],[1,1,1,1],[5,2,0,0],[4,3,0,1],[0,0,0,0]]

In [24]: y = [[el for el in l if el != 0] for l in x]

In [25]: y
Out[25]: [[1, 3], [1, 1, 1, 1], [5, 2], [4, 3, 1], []]

或者，

In [28]: [filter(None, l) for l in x]
Out[28]: [[1, 3], [1, 1, 1, 1], [5, 2], [4, 3, 1], []]

或使用functools：

In [32]: map(functools.partial(filter, None), x)
Out[32]: [[1, 3], [1, 1, 1, 1], [5, 2], [4, 3, 1], []]

Answer 2

使用list comprehensions：

>>> [[i for i in j if i] for j in x]
[[1, 3], [1, 1, 1, 1], [5, 2], [4, 3, 1], []]

另外，请注意，我依赖于这样一个事实，即整数仅在等于 时才int被提升为，并且它们不等于。值得考虑的是使用而不是明智的，因为前者更明确，因此更 Pythonic。false0true0if i != 0if i

Answer 3

您可以filter列出：

y = map(lambda l: filter(None, l), x)

或者您可以使用partial：

from functools import partial

y = map(partial(filter, None), x)

Answer 4

x = [[1,0,0,3],[1,1,1,1],[5,2,0,0],[4,3,0,1],[0,0,0,0]]

print x
print 'id(x) ==',id(x)
for sublist in x:
    print map(id,sublist)

for sublist in x:
    for i in xrange(len(sublist)-1,-1,-1):
        if sublist[i]==0:
            del sublist[i]

print
print x
print 'id(x) ==',id(x)
for sublist in x:
    print map(id,sublist)

使用此代码，您可以真正从初始列表中消除零，而无需创建另一个列表，也就是说，此代码在x

id(x)和map(id,sublist)的打印是为了显示这种修改方式：
身份，即子列表元素的地址（对于剩余的元素）和列表x本身，治疗前后都一样

如果列表很大，就地修改它可能会很有趣。
如果不是这种情况，您可以使用所有创建新列表对象的其他解决方案。

索引以相反的方式运行，xrange(len(sublist)-1,-1,-1)因为如果从左到右探索列表，则会给出错误的结果

结果

[[1, 0, 0, 3], [1, 1, 1, 1], [5, 2, 0, 0], [4, 3, 0, 1], [0, 0, 0, 0]]
id(x) == 18725984
[10021864, 10021876, 10021876, 10021840]
[10021864, 10021864, 10021864, 10021864]
[10021816, 10021852, 10021876, 10021876]
[10021828, 10021840, 10021876, 10021864]
[10021876, 10021876, 10021876, 10021876]

[[1, 3], [1, 1, 1, 1], [5, 2], [4, 3, 1], []]
id(x) == 18725984
[10021864, 10021840]
[10021864, 10021864, 10021864, 10021864]
[10021816, 10021852]
[10021828, 10021840, 10021864]
[]