Python之zip用法

Jack原创大约 3 分钟

Python之zip用法

zip() 是 Python 中的一个内置函数，它用于将多个可迭代对象（如列表、元组、字符串等）“打包”成一个个元组，然后返回由这些元组组成的迭代器。通过这个函数，可以方便地将两个或多个可迭代对象中的元素一一对应起来。下面是 zip() 函数的详细用法和示例。

1. 基本用法

zip() 的基本语法如下：

zip(*iterables)

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

zip() 会返回一个迭代器，其中每个元素是一个元组，包含了所有传入的可迭代对象中对应位置的元素。

示例：

# 将两个列表打包成一个列表的元组
list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
zipped = zip(list1, list2)
print(list(zipped))  # Output: [(1, 'a'), (2, 'b'), (3, 'c')]

2. 处理不同长度的可迭代对象

zip() 会根据最短的可迭代对象进行截断，也就是说，当最短的可迭代对象被耗尽时，zip() 函数将停止生成元组。

示例：

# 不同长度的列表
list1 = [1, 2, 3]
list2 = ['a', 'b']
zipped = zip(list1, list2)
print(list(zipped))  # Output: [(1, 'a'), (2, 'b')]

3. 处理多个可迭代对象

zip() 可以处理多个可迭代对象。

示例：

# 三个列表
list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
list3 = [True, False, True]
zipped = zip(list1, list2, list3)
print(list(zipped))  # Output: [(1, 'a', True), (2, 'b', False), (3, 'c', True)]

4. 解压缩

使用 zip() 和 * 运算符可以实现解压缩操作，将元组的列表转换回多个可迭代对象。

示例：

# 解压缩
zipped = [(1, 'a'), (2, 'b'), (3, 'c')]
list1, list2 = zip(*zipped)
print(list1)  # Output: (1, 2, 3)
print(list2)  # Output: ('a', 'b', 'c')

5. 结合 `enumerate()` 使用

可以结合 zip() 和 enumerate() 一起使用，方便地获取索引和值。

示例：

# 结合 enumerate 使用
list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
for index, (num, char) in enumerate(zip(list1, list2)):
    print(index, num, char)
# Output:
# 0 1 a
# 1 2 b
# 2 3 c

6. 处理字符串

zip() 可以用于字符串，将字符串中的字符一一对应起来。

示例：

# 处理字符串
str1 = "ABC"
str2 = "123"
zipped = zip(str1, str2)
print(list(zipped))  # Output: [('A', '1'), ('B', '2'), ('C', '3')]

7. 处理字典

可以使用 zip() 处理字典的键和值。

示例：

# 处理字典
keys = ['name', 'age', 'city']
values = ['Alice', 25, 'New York']
zipped = zip(keys, values)
dictionary = dict(zipped)
print(dictionary)  # Output: {'name': 'Alice', 'age': 25, 'city': 'New York'}

8. 用于生成器

zip() 可以与生成器结合使用，生成一一对应的元素对。

示例：

# 用于生成器
def generate_numbers():
    for i in range(1, 4):
        yield i

numbers = generate_numbers()
letters = ['A', 'B', 'C']
zipped = zip(numbers, letters)
print(list(zipped))  # Output: [(1, 'A'), (2, 'B'), (3, 'C')]

9. 结合列表推导式使用

zip() 可以与列表推导式结合使用，以便更方便地处理和生成数据。

示例：

# 结合列表推导式
list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
combined = [(num, char) for num, char in zip(list1, list2)]
print(combined)  # Output: [(1, 'a'), (2, 'b'), (3, 'c')]

注意事项

zip() 函数返回一个迭代器，聚合来自多个可迭代对象的元素。
结果的长度与最短的输入可迭代对象长度一致。
如果可迭代对象长度不一致，使用 itertools.zip_longest() 来填充缺失值。
常用于并行迭代多个序列。

写在最后

zip() 是一个非常强大的工具，可以方便地将多个可迭代对象的元素一一对应起来。无论是处理列表、元组、字符串还是字典，zip() 都提供了一种简洁而高效的方法。通过结合 enumerate()、列表推导式以及生成器，可以实现更复杂的数据处理逻辑。