Python標準函式庫itertools模組使用方法

簡介

官方描述：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
.
.

it​​ertools.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中文網其他相關文章！