python3.x zip用法详解

Posted 2022-12-05 bitcarmanlee

1.zip用法简介

在python 3.x系列中，zip方法返回的为一个zip object可迭代对象。

class zip(object):
    """
    zip(*iterables) --> zip object
    
    Return a zip object whose .__next__() method returns a tuple where
    the i-th element comes from the i-th iterable argument.  The .__next__()
    method continues until the shortest iterable in the argument sequence
    is exhausted and then it raises StopIteration.
    """

通过上面的注释，不难看出该迭代器的两个关键点：
1.迭代器的next方法返回一个元组，元组的第i个元素为各个输入参数的第i个元素。
2.迭代器的next方法，遇到输入序列中最短的那个序列迭代完毕，则会停止运行。

为了看清楚zip的效果，我们先看个最简单的例子

def fun0():
    a = ['a', 'b', 'c', 'd']
    b = ['1', '2', '3', '4']
    result = zip(a, b)
    print(type(result))
    try:
        while True:
            print(next(result))
    except StopIteration:
        pass

上面的代码，输出结果为

<class 'zip'>
('a', '1')
('b', '2')
('c', '3')
('d', '4')

首先可以看到的是，zip方法返回的，是一个zip对象。
zip对象是个迭代器，用next方法即可对其完成遍历。

当然我们也可以用for循环完成对zip对象的遍历。

def fun00():
    a = ['a', 'b', 'c', 'd']
    b = ['1', '2', '3', '4']
    result = zip(a, b)
    for ele in result:
        print(ele)

('a', '1')
('b', '2')
('c', '3')
('d', '4')

2.参数不等长进行截断

zip方法中，如果传入的参数不等长，则会进行截断，截断的时候会取最短的那个序列，超过最短序列长度的其他序列元素，则会被舍弃掉。

def fun0():
    a = ['a', 'b', 'c', 'd']
    b = ['1', '2', '3', '4', '5', '6']
    result = zip(a, b)
    try:
        while True:
            print(next(result))
    except StopIteration:
        pass

上述的方法如果运行，结果为

('a', '1')
('b', '2')
('c', '3')
('d', '4')

3.python3.x 与2.x中zip的不同

python3.x中，zip方法返回的是一个zip对象，本质是一个迭代器。而在python2.x中，zip方法直接返回一个list。
返回迭代器的好处在于，可以节省list占用的内存，只在有需要的时候再调用相关数据。

4.用zip方法构建字典

zip方法在实际中用途非常广泛，我们下面可以看几个实际中常用的例子。
zip方法可以用来构建字典。
字典包含两部分数据：key列表与value列表。如果我们现在有key序列与value序列，用zip方法可以很快构建一个字典。

def fun5():
    names = ['lili', 'lucy', 'tracy', 'larry']
    scores = [98, 10, 75, 90]
    my_dict = dict(zip(names, scores))
    print(my_dict)

'lili': 98, 'lucy': 10, 'tracy': 75, 'larry': 90

5.对多个序列的元素进行排序

排序也是日常工作中的常见需求，对多个序列进行排序而不破坏其元素的相对关系，也非常常见。下面我们来看一个常见的案例

还是以之前的数据为例
有names序列与scores序列，我们希望按照names进行排序，同时保持对应的scores数据。

def fun3():
    names = ['lili', 'lucy', 'tracy', 'larry']
    scores = [98, 10, 75, 90]
    data = sorted(list(zip(names, scores)), key=lambda x: x[0], reverse=False)
    print(data)

输出的结果为

[('larry', 90), ('lili', 98), ('lucy', 10), ('tracy', 75)]

如果我们希望按照分数逆序排，则可以按如下代码运行

def fun3():
    names = ['lili', 'lucy', 'tracy', 'larry']
    scores = [98, 10, 75, 90]
    data = sorted(list(zip(names, scores)), key=lambda x: x[1], reverse=True)
    print(data)

[('lili', 98), ('larry', 90), ('tracy', 75), ('lucy', 10)]

6.对多组数据进行计算

假设我们有3个序列，sales，costs，allowances。其中利润为销售额-成本+补贴，现在我们想求每组利润，就可以使用zip方法。

def fun4():
    sales = [10000, 9500, 9000]
    costs = [9000, 8000, 7000]
    allowances = [200, 100, 150]
    for sale, cost, allowance in zip(sales, costs, allowances):
        profit = sale - cost + allowance
        print(f"profit is: profit")

profit is: 1200
profit is: 1600
profit is: 2150

当然我们也可以使用for循环

def fun4():
    sales = [10000, 9500, 9000]
    costs = [9000, 8000, 7000]
    allowances = [200, 100, 150]
    for sale, cost, allowance in zip(sales, costs, allowances):
        profit = sale - cost + allowance
        print(f"profit is: profit")

    for i in range(len(sales)):
        profit = sales[i] - costs[i] + allowances[i]
        print(f"profit is: profit")

很明显zip方法比for循环还是要更直观，更简洁，更优雅。

7.*操作符进行解压

我们还可以使用*操作符对zip对象进行解压，效果是将zip object还原至原来的对象，效果就类似于压缩以后得解压。

def fun():
    a = ['a', 'b', 'c', 'd']
    b = ['1', '2', '3', '4']
    result = list(zip(a, b))
    print(result)
    zipobj = zip(a, b)
    a1, a2 = zip(*zipobj)
    print(list(a1))
    print(a2)

上面代码运行的结果为

[('a', '1'), ('b', '2'), ('c', '3'), ('d', '4')]
['a', 'b', 'c', 'd']
('1', '2', '3', '4')