展平列表 itertools 的列表

概述

在Python中，我们经常会遇到需要展平多层嵌套列表的情况，例如：

nested_list = [[1, 2, 3], [4, 5], [6, [7, 8]]]

如果我们希望将这个多层嵌套的列表展平为一维列表，常规的做法是使用递归或者循环来遍历列表中的元素。然而，这样做有时候会比较复杂，而且容易出错。因此，Python中的itertools模块提供了一种更加简单和高效的方法来展平多层嵌套列表。

itertools.chain()

itertools模块中的chain()函数可以将多个可迭代对象（包括列表、元组、集合和生成器等）合并为一个迭代器。我们可以利用这个函数来展平多层嵌套的列表。示例如下：

import itertools

nested_list = [[1, 2, 3], [4, 5], [6, [7, 8]]]

# 使用chain()函数将多个列表合并为一个迭代器，并转换为列表
flattened_list = list(itertools.chain(*nested_list))

print(flattened_list)    # [1, 2, 3, 4, 5, 6, [7, 8]]

可以看到，使用chain()函数展平列表之后，得到了一个包含所有元素的一维列表。但是，某些情况下，我们需要将嵌套的列表也展平为一维列表，这时候可以使用递归来解决。

itertools.chain.from_iterable()

itertools模块中的chain.from_iterable()函数可以将多个可迭代对象（包括列表、元组、集合和生成器等）合并为一个迭代器。和chain()函数不同的是，from_iterable()函数可以处理嵌套在可迭代对象中的可迭代对象，也就是多层嵌套的列表。示例如下：

import itertools

nested_list = [[1, 2, 3], [4, 5], [6, [7, 8]]]

# 使用chain.from_iterable()函数将多层嵌套的列表展平，并转换为列表
flattened_list = list(itertools.chain.from_iterable(nested_list))

print(flattened_list)    # [1, 2, 3, 4, 5, 6, 7, 8]

可以看到，使用chain.from_iterable()函数展平列表之后，得到了一个不包含嵌套列表的一维列表，这正是我们想要的结果。

总结

通过使用itertools模块中的chain()函数和chain.from_iterable()函数，我们可以高效地展平多层嵌套的列表，避免了使用递归或循环来遍历列表，能够极大地提高代码的可读性和性能。