导入
代码语言:javascript复制from itertools import *
无限迭代器
count()
作用: 无限生成数字,可设置步长。
格式:
代码语言:javascript复制count(start, step=1)
使用示例:
代码语言:javascript复制# 打印0,-2,-4,-6...
for i in count(0, step=-2):
print(i)
cycle()
作用: 无限生成字符。
格式:
代码语言:javascript复制cycle(string)
使用示例:
代码语言:javascript复制# 打印a,b,c,a,b...
for i in cycle("abc"):
print(i)
repeat()
作用: 无限打印元素。
格式:
代码语言:javascript复制repeat(item, n)
使用示例:
代码语言:javascript复制# 打印a,a,a,a,a...
for i in repeat("a"):
print(i)
# 打印b,b,b
for i in repeat("b", 3):
print(i)
有限迭代器
chain()
作用: 将迭代器合并。
格式:
代码语言:javascript复制chain(a1, a2, a3, ....)
使用示例:
代码语言:javascript复制# 输出a,b,c,1,2,3
for i in chain("abc", "123"):
print(i)
groupby()
作用: 把迭代器中相邻的元素按照key函数分组,当key=None时,把相邻的重复元素进行分组。
格式:
代码语言:javascript复制groupby(iterable, key=None)
使用示例:
代码语言:javascript复制# 对相邻的重复元素分组
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])...)
格式:
代码语言:javascript复制accumulate(iterable, func=sum)
使用示例:
代码语言:javascript复制# 分别对[: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循环。
格式:
代码语言:javascript复制product(p, q, …, repeat=1)
使用示例:
代码语言:javascript复制# 输出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()
作用: 排列,可指定长度。
格式:
代码语言:javascript复制permutations(p, r=len(p))
使用示例:
代码语言:javascript复制# 从序列中抽取两个元素的排列
for i in permutations("abcd", 2)
print(i)
combinations()
作用: 组合,无重复元素,可指定长度。
格式:
代码语言:javascript复制combinations(p, r=len(p))
使用示例:
代码语言:javascript复制# 从序列中抽取两个元素的组合
for i in combinations("abcd", 2)
print(i)
combinations_with_replacement()
作用: 组合,有重复元素,可指定长度。
格式:
代码语言:javascript复制combinations_with_replacement(p, r=len(p))
使用示例:
代码语言:javascript复制# 从序列中抽取两个元素的组合
for i in combinations_with_replacement("abcd", 2)
print(i)