python itertools模块chain函数功能详解 itertools.chain(*iterable)

Itertools模块， itertools提供了高效快捷的用于操作迭代对象的函数。通过使用这个模块，可以简化代码。

Itertools.chain语法

Itertools.chain(*iterables)

*代表接受可变的参数；

iterables，迭代对象（序列），可以使用for in 遍历的对象，包括list, string, dict。

1.Itertools.chain功能1：去除iterable里的内嵌iterable,如去除列表中的内嵌列表；

例子1.1 去除列表a中的内嵌元组;

import itertools as it
a=[(1, 'a'), (2, 'b'), (3, 'c')]
a_update=it.chain(*a)
print(type(a_update)) #itertools返回的对象是个类
print(list(a_update)) #转化成列表输出

程序运行结果为：

<class 'itertools.chain'>
[1, 'a', 2, 'b', 3, 'c']

例子1.2 去除列表b中的内嵌列表;

import itertools as it
b=[[1,2],[5,6],[8,9]]
b_update=it.chain(*b)
print(type(b_update)) #itertools返回的对象是个类
print(list(b_update)) #转化成列表输出

 程序运行结果为：

<class 'itertools.chain'>
[1, 2, 5, 6, 8, 9]

例子1.3 去除列表b中的内嵌字典; 

import itertools as it
c=[{1,2},{3,4},{5,6}]
c_update=it.chain(*c)
print(type(c_update)) #itertools返回的对象是个类
print(list(c_update)) #转化成列表输出

           程序运行结果为：

<class 'itertools.chain'>
[1, 2, 3, 4, 5, 6]

       

     2.Itertools.chain功能2：两个序列的组合

     例子2.1（类似python自带的+）

import itertools as it
a=[1,2,3]
b=[4,5,6]
print(a+b)
print(list(it.chain(*(a,b))))

           程序运行结果为：

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

      例子2.2结合zip函数进行配对问题；假如有列表a代表编号为1-5的5名女生，有列表b代表编号为a-e的5名男生，现在让5名男生依次插入5名女生中间，两队合成一队，如何用代码表示？

import itertools as it
a=[1,2,3,4,5] #女生
b=['a','b','c','d','e'] #男生
ab=zip(a,b)
final=it.chain(*ab)
print(list(final))

            程序运行结果为：

[1, 'a', 2, 'b', 3, 'c', 4, 'd', 5, 'e']

            例子2.3 合并字符串，并转换成列表（不同于python自带的+）

import itertools as it

m="abc"
n="def"
print(m+n)
print(list(it.chain(*(m,n))))

             程序运行结果为：

abcdef
['a', 'b', 'c', 'd', 'e', 'f']