python的函数itertools使用

1. 将二维列表合并成一维列表

具体API: itertools.chain()

1.1 案例

import itertools

tags_actual = [
    ['O', 'O', 'O', 'O', 'O', 'O', 'B-PER.NAM', 'I-PER.NAM', 'O', 'O', 'O', 'O', 'O', 'O', 'O'],
    ['O', 'B-PER.NAM', 'I-PER.NAM', 'I-PER.NAM', 'O', 'O', 'B-PER.NAM', 'I-PER.NAM', 'O', 'O']
]

tags_actual = itertools.chain(*tags_actual)
print(list(tags_actual))

结果输出：

['O', 'O', 'O', 'O', 'O', 'O', 'B-PER.NAM', 'I-PER.NAM', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'B-PER.NAM', 'I-PER.NAM', 'I-PER.NAM', 'O', 'O', 'B-PER.NAM', 'I-PER.NAM', 'O', 'O']

1.2 案例

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']