Python中的chain(*)函数|极客教程

Python中的chain(*)函数

1. 简介

在Python的itertools模块中，有一个非常方便的函数叫做chain(*iterables)，它接受一组可迭代对象作为参数，并将它们串联起来，返回一个迭代器。这个函数能够将多个可迭代对象合并成一个，从而方便地进行迭代操作。

2. 语法

itertools.chain(*iterables)

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

3. 功能

chain(*iterables)函数的作用是将多个可迭代对象拼接成一个串联的可迭代对象，可以直接用于遍历和访问所有元素。

4. 使用示例

下面我们来看一些使用chain()函数的示例代码：

4.1 示例1：链式合并多个列表

from itertools import chain

list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
list3 = [True, False]

merged_list = chain(list1, list2, list3)

for item in merged_list:
    print(item)

运行结果：

1
2
3
a
b
c
True
False

4.2 示例2：链式合并多个生成器

from itertools import chain

def gen1():
    yield 1
    yield 2

def gen2():
    yield 'a'
    yield 'b'

def gen3():
    yield True
    yield False

merged_gen = chain(gen1(), gen2(), gen3())

for item in merged_gen:
    print(item)

运行结果：

1
2
a
b
True
False

4.3 示例3：使用`chain.from_iterable()`函数

除了chain(*iterables)函数，itertools模块还提供了一个chain.from_iterable(iterables)函数，用于拼接一个可迭代对象的可迭代对象。下面是一个示例代码：

from itertools import chain

list_of_lists = [[1, 2], [3, 4], [5, 6]]

merged_list = chain.from_iterable(list_of_lists)

for item in merged_list:
    print(item)