Python学习之zip函数

灰太狼 2022-03-10 08:06 314阅读 0赞

# Python 学习之 zip 函数 #

## 问题的引出 ##

有时候，你可能想同时迭代两个序列。假设有下面两个列表：

`names = ['anne', 'beth', 'george', 'damon']`  
`ages = [12, 45, 32, 102]`  
如果要打印名字和对应的年龄，可以这样做：

>>> names = ['anne', 'beth', 'george', 'damon']
    >>> ages = [12, 45, 32, 102]
    >>> for i in range(len(names)): 
    ...     print(names[i], 'is', ages[i], 'years old')
    ... 
    anne is 12 years old
    beth is 45 years old
    george is 32 years old
    damon is 102 years old
    >>>

你可能觉得有点麻烦，尤其是第3行，写的代码有点多。有没有更好的办法呢？

## 并行迭代工具zip ##

Python 提供了一个很有用的并行迭代工具——内置函数 zip，它将两个序列“缝合”起来，并返回一个由元组组成的序列。返回值是一个适合迭代的对象，要查看其内容，可使用list将其转换为列表。

>>> list(zip(names, ages)) 
    [('anne', 12), ('beth', 45), ('george', 32), ('damon', 102)]

“缝合”后，可在循环中将元组解包。

>>> for name, age in zip(names, ages): 
    ...     print(name, 'is', age, 'years old') 
    ... 
    anne is 12 years old
    beth is 45 years old
    george is 32 years old
    damon is 102 years old
    >>>

函数zip可用于“缝合”任意数量的序列。需要指出的是，当序列的长度不同时，函数zip将在最短的序列用完后停止“缝合”。

>>> list(zip(range(5), range(100000000))) 
    [(0, 0), (1, 1), (2, 2), (3, 3), (4, 4)]

以下是其他的例子：

>>> a = [1,2,3]
    >>> b = [4,5,6]
    >>> c = [4,5,6,7,8]
    >>> zipped = zip(a,b)     # 返回一个对象
    >>> zipped
    <zip object at 0x103abc288>
    >>> list(zipped)          # 用 list() 转换为列表
    [(1, 4), (2, 5), (3, 6)]
    >>> list(zip(a,c))        # 元素个数与最短的列表一致
    [(1, 4), (2, 5), (3, 6)]
    >>> list(zip(a,b,c))      # 缝合3个序列
    [(1, 4, 4), (2, 5, 5), (3, 6, 6)]
    >>>

也可以把列表和元组缝合在一起。

>>> a = (1,2,3)
    >>> b = [4,5,6]
    >>> list(zip(a,b))
    [(1, 4), (2, 5), (3, 6)]

## `zip(*)`的用法 ##

操作符`*`与zip函数配合可以实现与zip相反的功能，即将“缝合”的序列拆成多个元组。

>>> a = [1,2]
    >>> b = [3,4]
    >>> c = ['a','b']
    >>> zipped = zip(a,b,c)    # 缝合
    >>> a1, a2, a3 = zip(*zipped)   # 拆包
    >>> a1
    (1, 2)
    >>> a2
    (3, 4)
    >>> a3
    ('a', 'b')
    >>> b1,b2,b3 = zip(*zip(a,b,c))
    >>> b1
    (1, 2)
    >>> b2
    (3, 4)
    >>> b3
    ('a', 'b')

## 参考资料 ##

\[1\] [http://www.runoob.com/python3/python3-func-zip.html][http_www.runoob.com_python3_python3-func-zip.html]

\[2\]《Python基础教程》人民邮电出版社，第3版

[http_www.runoob.com_python3_python3-func-zip.html]: http://www.runoob.com/python3/python3-func-zip.html