Python中itertools简介使用介绍

Python中Itertools简介和使用

简介

Python中的Itertools模块是一个提供有用的迭代器函数的模块。Itertools模块实现了很多有用的迭代器，这些迭代器可以用于完成很多任务，包括高效的循环，排列组合等。

安装

itertools模块是Python自带的标准库，无需手动安装。直接import itertools即可。

用法

1. itertools.chain()

itertools.chain()函数接受一个或多个可迭代对象，并把所有可迭代对象串联起来返回一个新的迭代器，可以用于便捷地组合多个迭代器。

import itertools

a = [1, 2, 3]
b = [4, 5, 6]
c = [7, 8, 9]

# 使用 itertools.chain()合并多个列表
d = itertools.chain(a, b, c)
for i in d:
    print(i, end=' ')
# output: 1 2 3 4 5 6 7 8 9

2. itertools.combinations()

itertools.combinations()函数用于计算组合。它接受两个参数，一个是可迭代对象，另一个是要选多少个元素作为一个组合。

import itertools

a = [1, 2, 3, 4]
for i in itertools.combinations(a, 2):
    print(i, end=', ')

# output: (1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4),

函数返回一个迭代器，每个元素都是由原始可迭代对象中选定的元素组成的元组。

注意事项

itertools生成的是迭代器，需要用for循环或者list()函数转换成列表再操作，直接使用print()打印不会有输出。

总结

Python中的itertools提供了很多强大的工具，可以大大提高我们编码时的效率。我们可以非常方便地使用itertools中的函数来完成很多常见的任务，例如组合、过滤、循环等。值得我们去深入了解和掌握。