Python中itertools详解

目录

• Python中itertools模块
  • 一、 简介
  • 二、 使用介绍
    • 1、 常用迭代器
      • 1.1 chain
      • 1.2 groupby
    • 2、 无穷迭代器
      • 2.1 count
      • 2.2 cycle
      • 2.3 repeat
    • 3、排列组合迭代器
      • 3.1 product
      • 3.2 permutations
      • 3.3 combinations
      • 3.4 combinations_with_replacement

Python中itertools模块

一、 简介

itertools是python内置的模块，使用简单且功能强大

官方文档地址：https://docs.python.org/zh-cn/3/library/itertools.html

itertools模块标准化了一个快速、高效利用内存的核心工具集，这些工具本身或组合都很有用。它们一起形成了“迭代器代数”，这使得在纯Python中有可能创建简洁又高效的专用工具。

同时，itertools模块是python的内置库，我们可以直接使用，不需要进行额外的安装

这里讲解一些常用的函数，其余函数请到官方文档查看学习。

二、 使用介绍

1、 常用迭代器

1.1 chain

函数介绍：chain可以把一组迭代对象串联起来，形成一个更大的迭代器

语法：(class) chain(*iterables: Iterable[str])

import itertools
l1 = ["A", "B"]
l2 = ["C", "D"]
print(list(itertools.chain(l1, l2)))  # 将两个列表展开，当然，也可以用到一些可迭代对象里面去，比如说字符串等，返回一个可迭代对象

1.2 groupby

函数介绍：groupby 把迭代器中相邻的重复元素（key）挑出来放一起

语法：(class) groupby(iterable: Iterable[_T1@__new__], key: None = ...)

import itertools
for key, value in itertools.groupby("hello world! My name is Steve Anthony"):
    print(key, list(value))
    
"""结合key使用"""
import itertools
data = [
    (1, "Make", 93),
    (1, "Jack", 100), 
    (2, "Lucy", 90)
]
for key, value in itertools.groupby(data, key=lambda student: student[0]):
    print(key, list(value))

2、 无穷迭代器

2.1 count

函数作用：生成无界限序列，count(start=0, step=1) ，示例从100开始，步长为2，循环10，打印对应值；必须手动break，count()会一直循环。

语法：count(start: int, [step: _Step = ...])

import itertools

for i in itertools.count(10):
    print(i)=

2.2 cycle

函数作用：对可迭代容器里面的元素进行无限循环

语法：(class) cycle(__iterable: Iterable[int], /)

import itertools

for i in itertools.cycle(range(10)):
    print(i)

2.3 repeat

函数作用：对可迭代容器里面的元素重复times次

语法：repeat(object: range, times: int)

import itertools

for i in itertools.repeat(range(10), 2):
    print(i)

3、排列组合迭代器

3.1 product

函数作用：返回笛卡尔积，相当于嵌套的for循环，重复repeat次

语法：product(*iterables: Iterable, repeat: int = 1)

import itertools

for i in itertools.product([1, 2], ["A", "B", "C"]):
    print(i)

3.2 permutations

函数作用：生成一个长度为r的元组，元组中存放所有可能的排列，无重复元素

语法：(class) permutations(iterable: Iterable, r: int | None = ...)

import itertools

for i in itertools.permutations([1, 2, 3], 3):  # r的默认值为2
    print(i)

3.3 combinations

函数作用：生成一个长度为r的元组，有序，并且无重复元素

语法：(class) combinations(iterable: Iterable[_T@__new__], r: Literal[2])

import itertools

for i in itertools.combinations([1, 2, 3], 3):
    print(i)

3.4 combinations_with_replacement

函数作用：生成一个长度为r的元组，有序，有重复元素

import itertools

for i in itertools.combinations_with_replacement([1, 2, 3], 3):
    print(i)

这些是一些常用的迭代器函数，如果还想要更加深入的了解的话，可以去官方文档查看！