标题: Python标准库之 Itertools模块 [打印本页] 作者: w8484855@qq.com 时间: 2017-8-16 17:19 标题: Python标准库之 Itertools模块 本文和大家分享的主要是python标准库中Itertools模块相关内容,一起来看看吧,希望对大家学习python有所帮助。
简介
官方描述:Functional tools for creating and using iterators.即用于创建高效迭代器的函数。
itertools.chain(*iterable)
将多个序列作为一个单独的序列返回。 例如:
import itertoolsfor each in itertools.chain('i', 'love', 'python'):
print each
输出:
i
l
o
v
e
p
y
t
h
o
n
itertools.combinations(iterable, r)
返回指定长度的”组合” 例如:
import itertoolsfor each in itertools.combinations('abc', 2):
print each
输出:
('a', 'b')
('a', 'c')
('b', 'c')
itertools.combinations_with_replacement(iterable, r)
返回指定长度的“组合”,组合内元素可重复 例如:
import itertoolsfor each in itertools.combinations_with_replacement('abc', 2):
print each
输出:
('a', 'a')
('a', 'b')
('a', 'c')
('b', 'b')
('b', 'c')
('c', 'c')
itertools.product(*iterable[,repeat])
返回指定长度的所有组合,可理解为笛卡尔乘积 例如:
import itertoolsfor each in itertools.product('abc', repeat=2):
print each
(‘a’, ‘a’) (‘a’, ‘b’) (‘a’, ‘c’) (‘b’, ‘a’) (‘b’, ‘b’) (‘b’, ‘c’) (‘c’, ‘a’) (‘c’, ‘b’) (‘c’, ‘c’)
itertools.premutations(iteravle[,r])
返回长度为r的排列 例如:
import itertoolsfor value in itertools.permutations('abc', 2):
print value
输出:
('a', 'b')
('a', 'c')
('b', 'a')
('b', 'c')
('c', 'a')
('c', 'b')
itertools.compress(data,selector)
返回selector为True的data对应元素 例如:
import itertoolsfor each in itertools.compress('abcd', [1, 0, 1, 0]):
print each
输出:
a
c
itertools.count(start=0,step=1)
返回以start开始,step递增的序列,无限递增 例如:
import itertoolsfor each in itertools.count(start=0, step=2):
print each
输出:
123
.
.
itertools.cycle(iterable)将迭代器进行无限迭代 例如:
import itertoolsfor each in itertools.cycle('ab'):
print each
输出:
a
b
a
b
.
.
itertools.dropwhile(predicate, iterable)
直到predicate为真,就返回iterable后续数据, 否则drop掉 例如:
import itertoolsfor each in itertools.dropwhile(lambda x: x<5, [2,1,6,8,2,1]):
print each
输出:
itertools.groupby(iterable[,key])
返回一组(key,itera),key为iterable的值,itera为等于key的所有项 例如:
import itertoolsfor key, vale in itertools.groupby('aabbbc'):
print key, list(vale)
输出:
a ['a', 'a']
b ['b', 'b', 'b']
c ['c']
itertools.ifilter(predicate, iterable)
返回predicate结果为True的元素迭代器,如果predicate为None,则返回所有iterable中为True的项 例如:
import itertoolsfor value in itertools.ifilter(lambda x: x % 2, range(10)):
print value
输出:
itertools.ifilterfasle(predicate,iterable)
返回predicate为False的元素,如果predicate为None,则返回所有iterable中为False的项 例如:
import itertoolsfor value in itertools.ifilterfalse(lambda x: x % 2, range(10)):
print value
输出:
itertools.imap(function,*iterables)
相当于迭代器方式的map() 例如:
import itertoolsfor value in itertools.imap(lambda x, y: x+y, (1,2,3), (4,5,6)):
print value
输出:
itertools.islice(iterable, start,stop[,step])
相当于迭代器方式的切片操作 例如:
import itertoolsfor value in itertools.islice('abcdefg', 1, 4, 2):
print value
输出:
b
d
itertools.repeat(object,[,times])
不停的返回object对象,如果指定了times,则返回times次 例如:
import itertoolsfor value in itertools.repeat('a', 2):
print value
输出:
a
a
itertools.starmap(function,iterable)
返回function(iter)的值,iter为iterable的元素 例如:
import itertoolsfor value in itertools.starmap(lambda x, y: x * y, [(1, 2), (3, 4)]):
print value
输出:
itertools.takewhile(predicate,iterable)
如果predicate为真,则返回iterable元素,如果为假则不再返回,break. 例如:
import itertoolsfor value in itertools.takewhile(lambda x: x < 5, [1, 3, 5, 6]):
print value
输出:
1
3