python2和3中zip的差异(坑)

发布时间:2019-09-16 07:20:18编辑:auto阅读(1699)

    最近用python写个东西,里面用到了一个zip,内存飙升,甚是费解,查阅资料后发现zip在python2和3中的机制不同

    class Foo(object):
        def __init__(self):
            self.items = range(3)
        def __getitem__(self, i):
            print('__getitem__', i)
            return self.items[i]
    foo1 = Foo()
    foo2 = Foo()
    print(zip(foo1, foo2))
    # for s1, s2 in zip(foo1, foo2):
    #     print(s1, s2)

    python2

    # 输出:
    # 提前已经遍历了一遍,组成新的列表了
    ('__getitem__', 0)
    ('__getitem__', 0)
    ('__getitem__', 1)
    ('__getitem__', 1)
    ('__getitem__', 2)
    ('__getitem__', 2)
    ('__getitem__', 3)
    [(0, 0), (1, 1), (2, 2)]

    python3

    # 输出:
    <zip object at 0x7f96933f9648>

    可以看到python2中是先把2个foo**全部遍历一下组成一个新的列表,十分耗内存**
    而python3是返回一个迭代器,遍历zip(foo1, foo2)时才会去取foo1和foo2的每个元素,不费内存


关键字