from itertools import *
作用: 无限生成数字,可设置步长。
格式:
count(start, step=1)
使用示例:
# 打印0,-2,-4,-6...
for i in count(0, step=-2):
print(i)
作用: 无限生成字符。
格式:
cycle(string)
使用示例:
# 打印a,b,c,a,b...
for i in cycle("abc"):
print(i)
作用: 无限打印元素。
格式:
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(a1, a2, a3, ....)
使用示例:
# 输出a,b,c,1,2,3
for i in chain("abc", "123"):
print(i)
作用: 把迭代器中相邻的元素按照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)
作用: 计算迭代器,如果不指定参数函数,将会采取默认的求和函数。(实际上是对序列的顺序子序列使用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)
作用: 笛卡尔积,相当于嵌套的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(p, r=len(p))
使用示例:
# 从序列中抽取两个元素的排列
for i in permutations("abcd", 2)
print(i)
作用: 组合,无重复元素,可指定长度。
格式:
combinations(p, r=len(p))
使用示例:
# 从序列中抽取两个元素的组合
for i in combinations("abcd", 2)
print(i)
作用: 组合,有重复元素,可指定长度。
格式:
combinations_with_replacement(p, r=len(p))
使用示例:
# 从序列中抽取两个元素的组合
for i in combinations_with_replacement("abcd", 2)
print(i)