好资源导航 » 文章资讯 » Pythonitertools.takewhile

Pythonitertools.takewhile

2024-04-13 08:10:03 435

示例

itertools.takewhile使您可以从序列中取出项目，直到条件首次变为False。

def is_even(x):
    return x % 2 == 0


lst = [0, 2, 4, 12, 18, 13, 14, 22, 23, 44]
result = list(itertools.takewhile(is_even, lst))

print(result)

这输出[0,2,4,12,18]。

请注意，违反谓词的第一个数字（即：返回布尔值的函数）is_even是13。一旦takewhile遇到False为给定谓词产生的值，它就会爆发。

在产生的输出通过takewhile类似于从下面的代码所产生的输出。

def takewhile(predicate, iterable):
    for x in iterable:
        if predicate(x):
            yield x
        else:
            break

注意：由产生的结果的串联takewhile和dropwhile产生原始可迭代的结果。

result=list(itertools.takewhile(is_even,lst))+list(itertools.dropwhile(is_even,lst))