itertools
导入
from itertools import *无限迭代器
count()
作用： 无限生成数字，可设置步长。
格式：
count(start, step=1)使用示例：
# 打印0，-2，-4，-6...
for i in count(0, step=-2):
print(i)cycle()
作用： 无限生成字符。
格式：
cycle(string)使用示例：
# 打印a，b，c，a，b...
for i in cycle("abc"):
print(i)repeat()
作用： 无限打印元素。
格式：
repeat(item, n)使用示例：
# 打印a，a，a，a，a...
for i in repeat("a"):
print(i)
# 打印b，b，b
for i in repeat("b", 3):
print(i)有限迭代器
chain()
作用： 将迭代器合并。
格式：
chain(a1, a2, a3, ....)使用示例：
# 输出a，b，c，1，2，3
for i in chain("abc", "123"):
print(i)groupby()
作用： 把迭代器中相邻的元素按照key函数分组，当key=None时，把相邻的重复元素进行分组。
格式：
groupby(iterable, key=None)使用示例：
# 对相邻的重复元素分组
for key group in chain("aaabbbcccdd"):
print(key, group)
# 对大写相同的相邻元素分组
for key group in chain("AaaBbbcCcDd", lambda x: x.upper()):
print(key, group)accumulate()
作用： 计算迭代器，如果不指定参数函数，将会采取默认的求和函数。（实际上是对序列的顺序子序列使用func函数并返回结果，比如说max(iterable[:1])，max(iterable[:2])，max(iterable[:3])...）
格式：
accumulate(iterable, func=sum)使用示例：
# 分别对[:0]，[:1]，[:2]求和
for i in accumulate([1, 2, 3, 4, 5, 6], func=sum):
print(i)
# 求前i个元素的中最大值
for i in accumulate([1, 2, 3, 4, 5, 6], func=max):
print(i)组合迭代器
product()
作用： 笛卡尔积，相当于嵌套的for循环。
格式：
product(p, q, …, repeat=1)使用示例：
# 输出ac，ad，bc，bd
for i in product("ab", "cd", …, repeat=1):
print(i)
# 输出aa，ab，ba，bb
for i in product("ab"repeat=2):
print(i)permutations()
作用： 排列，可指定长度。
格式：
permutations(p, r=len(p))使用示例：
# 从序列中抽取两个元素的排列
for i in permutations("abcd", 2)
print(i)combinations()
作用： 组合，无重复元素，可指定长度。
格式：
combinations(p, r=len(p))使用示例：
# 从序列中抽取两个元素的组合
for i in combinations("abcd", 2)
print(i)combinations_with_replacement()
作用： 组合，有重复元素，可指定长度。
格式：
combinations_with_replacement(p, r=len(p))使用示例：
# 从序列中抽取两个元素的组合
for i in combinations_with_replacement("abcd", 2)
print(i)本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2023-08-08，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录