一文详解Python中的zip函数

目录

• 初识zip
• 基本用法
• 解压缩
• 处理不等长序列
• 使用itertools.zip_longest处理不等长序列
• zip函数的灵活性
• 与*运算符结合

在Python中，处理数据时经常需要同时遍历多个序列。zip函数提供了一种简洁的方式来组合这些序列，它用于将多个可迭代对象（如列表、元组等）的元素配对，创建一个新的迭代器，该迭代器的元素是由输入的可迭代对象中并行的元素组成的元组。使得我们可以轻松地将多个序列的元素配对处理。

初识zip

zip函数可以接受任意数量的可迭代对象作为参数。

zip(*iterables)

iterables：一个或多个可迭代对象。

基本用法

numbers = [1, 2, 3]
colors = ['red', 'blue', 'green']

zipped_pairs = zip(numbers, colors)

zipped_pairs是一个迭代器，包含了numbers和colors中元素的配对，即使numbers和colors类型（一个列表，一个元祖）不同也可以。

<zip object at 0x7f9df0d9c190>

要查看具体内容，可以将其转换为列表。

list(zipped_pairs)

输出结果：

[(1, 'red'), (2, 'blue'), (3', 'green')]

解压缩

zip对象支持解压缩，可以直接在for循环中使用，方便迭代。

for number, color in zip(numbers, colors):
    print(f"The number {number} is the color {color}")

处理不等长序列

当尝试zip处理的两个可迭代对象是不等长的序列时，较短的序列会限制生成的元组数量。

short_numbers = [1, 2]
long_colors = ['red', 'blue', 'green', 'yellow']

zipped = zip(short_numbers, long_colors)
list(zipped)

输出结果：

[(1, 'red'), (2, 'blue')]

使用itertools.zip_longest处理不等长序列

对于不等长的序列，如果需要处理到最长序列的末尾，可以使用itertools.zip_longest。

import itertools

zipped_longest = itertools.zip_longest(short_numbers, long_colors)
list(zipped_longest)

输出结果：

[(1, 'red'), (2, 'blue'), (None, 'green'), (None, 'yellow')]

itertools.zip_longest允许你指定一个fillvalue参数，用于填充缺失的值，默认为None。

zip函数的灵活性

zip函数不仅适用于列表和元组，它还可以处理任何可迭代对象，包括字符串、字典、集合等。

与*运算符结合

zip与*运算符结合，可以快速生成字典。

keys = ['a', 'b', 'c']
values = [1, 2, 3]

my_dict = dict(zip(keys, values))
print(my_dict)

输出结果：

{'a': 1, 'b': 2, 'c': 3}