268

我有一个包含 20000 个列表的列表。我使用每个列表的第三个元素作为标志。只要至少一个元素的标志为0，我就想在这个列表上做一些操作，就像：

my_list = [["a", "b", 0], ["c", "d", 0], ["e", "f", 0], .....]

一开始，所有标志都是 0。我使用 while 循环检查是否至少一个元素的标志为 0：

def check(list_):
    for item in list_:
        if item[2] == 0:
            return True
    return False

如果check(my_list)返回True，那么我将继续处理我的列表：

while check(my_list):
    for item in my_list:
        if condition:
            item[2] = 1
        else:
            do_sth()

实际上，我想在迭代时删除 my_list 中的一个元素，但不允许在迭代时删除项目。

原始 my_list 没有标志：

my_list = [["a", "b"], ["c", "d"], ["e", "f"], .....]

由于在迭代元素时无法删除元素，因此我发明了这些标志。但是my_list包含许多项目，并且while循环在每个循环中读取所有项目，并且for消耗大量时间！你有什么建议吗？

4

5 回答 5

482

这里最好的答案是使用all()，这是这种情况的内置函数。我们将其与生成器表达式相结合，以干净有效地产生您想要的结果。例如：

>>> items = [[1, 2, 0], [1, 2, 0], [1, 2, 0]]
>>> all(flag == 0 for (_, _, flag) in items)
True
>>> items = [[1, 2, 0], [1, 2, 1], [1, 2, 0]]
>>> all(flag == 0 for (_, _, flag) in items)
False

请注意，all(flag == 0 for (_, _, flag) in items)它直接等同于all(item[2] == 0 for item in items)，在这种情况下阅读起来会更好一些。

而且，对于过滤器示例，列表理解（当然，您可以在适当的情况下使用生成器表达式）：

>>> [x for x in items if x[2] == 0]
[[1, 2, 0], [1, 2, 0]]

如果您想检查至少一个元素是否为 0，则更好的选择是使用any()更具可读性的选项：

>>> any(flag == 0 for (_, _, flag) in items)
True
于 2012-05-19T15:04:53.403 回答
23

如果要检查列表中的任何项目是否违反条件，请使用all：

if all([x[2] == 0 for x in lista]):
    # Will run if all elements in the list has x[2] = 0 (use not to invert if necessary)

要删除所有不匹配的元素，请使用filter

# Will remove all elements where x[2] is 0
listb = filter(lambda x: x[2] != 0, listb)
于 2012-05-19T14:55:19.923 回答
7

您可以像这样使用 itertools 的 takewhile，一旦满足使您的语句失败的条件，它将停止。相反的方法是 dropwhile

for x in itertools.takewhile(lambda x: x[2] == 0, list)
    print x
于 2012-05-19T14:54:07.593 回答
1

这种方式比使用更灵活all()：

my_list = [[1, 2, 0], [1, 2, 0], [1, 2, 0]]
all_zeros = False if False in [x[2] == 0 for x in my_list] else True
any_zeros = True if True in [x[2] == 0 for x in my_list] else False

或更简洁地说：

all_zeros = not False in [x[2] == 0 for x in my_list]
any_zeros = 0 in [x[2] for x in my_list]
于 2015-11-08T09:56:55.003 回答
0

另一种使用方式itertools.ifilter。这检查真实性和过程（使用lambda）

样本-

for x in itertools.ifilter(lambda x: x[2] == 0, my_list):
    print x
于 2015-11-13T05:56:38.037 回答