Python标准库itertools模块使用方法

高洛峰
发布: 2017-03-21 11:27:54
原创
1834 人浏览过

简介

官方描述:Functional tools for creating and using iterators.即用于创建高效迭代器的函数。

itertools.chain(*iterable)

将多个序列作为一个单独的序列返回。
例如:

import itertools
for 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 itertools
for each in itertools.combinations('abc', 2):
    print each
登录后复制

输出:

('a', 'b')
('a', 'c')
('b', 'c')
登录后复制

itertools.combinations_with_replacement(iterable, r)

返回指定长度的“组合”,组合内元素可重复
例如:

import itertools
for 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 itertools
for 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 itertools
for 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 itertools
for each in itertools.compress('abcd', [1, 0, 1, 0]):
    print each
登录后复制

输出:

a
c
登录后复制

itertools.count(start=0,step=1)

返回以start开始,step递增的序列,无限递增
例如:

import itertools
for each in itertools.count(start=0, step=2):
    print each
登录后复制

输出:

1
2
3
.
.
登录后复制

itertools.cycle(iterable)

将迭代器进行无限迭代
例如:

import itertools
for each in itertools.cycle('ab'):
    print each
登录后复制

输出:

a
b
a
b
.
.
登录后复制

itertools.dropwhile(predicate, iterable)

直到predicate为真,就返回iterable后续数据, 否则drop掉
例如:

import itertools
for each in itertools.dropwhile(lambda x: x<5, [2,1,6,8,2,1]):
    print each
登录后复制

输出:

6
8
2
1
登录后复制

itertools.groupby(iterable[,key])

返回一组(key,itera),key为iterable的值,itera为等于key的所有项
例如:

import itertools
for 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 itertools
for value in itertools.ifilter(lambda x: x % 2, range(10)):
    print value
登录后复制

输出:

1
3
5
7
9
登录后复制

itertools.ifilterfasle(predicate,iterable)

返回predicate为False的元素,如果predicate为None,则返回所有iterable中为False的项
例如:

import itertools
for value in itertools.ifilterfalse(lambda x: x % 2, range(10)):
    print value
登录后复制

输出:

0
2
4
6
8
登录后复制

itertools.imap(function,*iterables)

相当于迭代器方式的map()
例如:

import itertools
for value in itertools.imap(lambda x, y: x+y, (1,2,3), (4,5,6)):
    print value
登录后复制

输出:

5
7
9
登录后复制

itertools.islice(iterable, start,stop[,step])

相当于迭代器方式的切片操作
例如:

import itertools
for value in itertools.islice('abcdefg', 1, 4, 2):
    print value
登录后复制

输出:

b
d
登录后复制

itertools.repeat(object,[,times])

不停的返回object对象,如果指定了times,则返回times次
例如:

import itertools
for value in itertools.repeat('a', 2):
    print value
登录后复制

输出:

a
a
登录后复制

itertools.starmap(function,iterable)

返回function(iter)的值,iter为iterable的元素
例如:

import itertools
for value in itertools.starmap(lambda x, y: x * y, [(1, 2), (3, 4)]):
    print value
登录后复制

输出:

2
12
登录后复制

itertools.takewhile(predicate,iterable)

如果predicate为真,则返回iterable元素,如果为假则不再返回,break.
例如:

import itertools
for value in itertools.takewhile(lambda x: x < 5, [1, 3, 5, 6]):
    print value
登录后复制

输出:

1
3
登录后复制

以上是Python标准库itertools模块使用方法的详细内容。更多信息请关注PHP中文网其他相关文章!

相关标签:
来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板