Python – Itertools.chain.from_iterable()

在Python中，使用itertools模块的chain.from_iterable()函数可以帮助我们将嵌套的迭代器内容合并在一起，并生成一个单一的迭代器。此功能对于数字列表，字符串列表和其他可迭代对象尤其有用。

使用方法

要使用chain.from_iterable()函数，需要先导入itertools模块。下面是一个使用chain.from_iterable()函数的示例代码：

import itertools

nested_list = [[1, 2], [3, 4], [5, 6]]
flat_list = list(itertools.chain.from_iterable(nested_list))
print(flat_list)

输出：

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

在此示例中，我们有一个嵌套的列表[[1, 2], [3, 4], [5, 6]]。通过使用itertools.chain.from_iterable()函数，我们将其展平成一个单一的列表。最后，我们使用list()函数将展平的迭代器转换为列表。

另一个示例

以下是另一个示例，此示例将字符串列表展平：

import itertools

nested_list = [['hello', 'world'], ['foo', 'bar'], ['baz']]
flat_list = list(itertools.chain.from_iterable(nested_list))
print(flat_list)

输出：

['hello', 'world', 'foo', 'bar', 'baz']

总结

chain.from_iterable()函数可以帮助我们将嵌套的迭代器中的内容合并在一起，并生成一个单一的迭代器。这对于数字列表，字符串列表和其他可迭代对象尤其有用。