美文网首页Python
itertools.chain() --在不同的容器中进行迭代

itertools.chain() --在不同的容器中进行迭代

作者: cook__ | 来源:发表于2018-10-03 19:37 被阅读39次

    问题:我们需要对许多对象执行相同的操作,但是这些对象包含在不同的容器内,而我们希望可以避免写出嵌套循环,保持代码的可读性

    解决方案:使用itertools.chain()可以解决这个问题,它接受一系列可迭代对象作为输入并返回一个迭代器(实际上是在对多个容器进行迭代)
    import itertools
    a = [1, 2, 3, 4]
    b = ['x', 'y', 'z']
    for x in itertools.chain(a, b):
        print(x)
        
    1
    2
    3
    4
    x
    y
    z
    
    总结:

    1、itertools.chain()可接受一个或多个可迭代对象作为参数,然后会创建一个迭代器,该迭代器可连续访问并返回提供的每个可迭代对象中的元素;
    2、如果输入的序列很大,在内存使用上chain()就会高效很多,而且当可迭代对象之间不是同一种类型时也可以轻松适用。

    相关文章

      网友评论

        本文标题:itertools.chain() --在不同的容器中进行迭代

        本文链接:https://www.haomeiwen.com/subject/irctaftx.html