python - itertools 模塊

相關文檔

文檔 連接 

pymotw 連接 

無限迭代器

itertools.count()

說明

生成一個無限迭代的數字隊列, 只有進行 

參數 

其實數字以及步幅

start, [step]

返回值 

start, start+step, start+2*step, ...

示例

count(10) --> 10 11 12 13 14 ...

from itertools import *
import time


c = count(10,5)
print c


for i in c:
    time.sleep(0.5)
    print i

"""
count(10, 5)
10
15
20
25
30
35
40
45
50
55

Process finished with exit code -1
"""

詳細示例

itertools.cycle()

說明

傳入一個可迭代對象而後循環遍歷此對象

參數 

p

返回值 

p0, p1, ... plast, p0, p1, ...

示例

cycle('ABCD') --> A B C D A B C D ...

from itertools import *
import time

c = cycle([1, 2, 3])
print c

for i in c:
    time.sleep(0.5)
    print i

"""
<itertools.cycle object at 0x0000000002A42088>
1
2
3
1
2
3
1
2

Process finished with exit code -1

"""

詳細示例

itertools.repeat()

說明

傳入一個對象, 以及數字, 遍歷此對象指定數字次數

參數 

elem [,n]

返回值 

elem, elem, elem, ... endlessly or up to n times

示例

repeat(10, 3) --> 10 10 10

from itertools import *
import time

c = repeat(3, 5)
print c

for i in c:
    time.sleep(0.5)
    print i

"""
repeat(3, 5)
3
3
3
3
3

Process finished with exit code 0
"""

詳細示例

處理輸入序列迭代器

itertools.chain()

說明

傳入多個序列, 遍歷時將多個序列串聯一塊兒進行

參數 

p, q, ...

返回值 

p0, p1, ... plast, q0, q1, ...

示例

chain('ABC', 'DEF') --> A B C D E F

from itertools import *

c = chain(range(5), range(3))
print c

for i in c:
    print i

"""
<itertools.chain object at 0x0000000002A77FD0>
0
1
2
3
4
0
1
2

Process finished with exit code 0r
"""

詳細示例

itertools.compress()

說明

參數傳入一個序列以及一個選擇列表，對數據序列基於選擇列表進行篩選

參數 

data, selectors 

返回值 

(d[0] if s[0]), (d[1] if s[1]), ...         

示例

compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F

from itertools import *

c = compress("hahahahahaha", [1, 0, 1, 0, 1, 0, 1, 0])
print c

for i in c:
    print i

"""
<itertools.compress object at 0x0000000002B01048>
h
h
h
h

Process finished with exit code 0
"""

詳細示例

itertools.dropwhile()

說明

在條件爲false以後的第一次, 返回迭代器中剩下來的項.

參數 

pred, seq

返回值 

 seq[n], seq[n+1], starting when pred fails

示例

 dropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1

def dropwhile(predicate, iterable):
    # dropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1
    iterable = iter(iterable)
    for x in iterable:
        if not predicate(x):
            yield x
            break
    for x in iterable:
        yield x

from itertools import *
  
 
def should_drop(x):
    print 'Testing:', x
    return (x<1)
 
for i in dropwhile(should_drop, [ -1, 0, 1, 2, 3, 4, 1, -2 ]):
    print 'Yielding:', i
 
Testing: -1
Testing: 0
Testing: 1
Yielding: 1
Yielding: 2
Yielding: 3
Yielding: 4
Yielding: 1
Yielding: -2

詳細示例

itertools.groupby()

說明

按照keyfunc函數對序列每一個元素執行後的結果分組(每一個分組是一個迭代器), 返回這些分組的迭代器

參數 

iterable[, keyfunc]

返回值 

sub-iterators grouped by value of keyfunc(v)

示例

chain('ABC', 'DEF') --> A B C D E F

 

 

 

 

迭代器       　　參數            結果                                        例子
ifilter()   pred, seq           elements of seq where pred(elem) is True    ifilter(lambda x: x%2, range(10)) --> 1 3 5 7 9
ifilterfalse()  pred, seq       elements of seq where pred(elem) is False   ifilterfalse(lambda x: x%2, range(10)) --> 0 2 4 6 8
islice()    seq, [start,] stop [, step] elements from seq[start:stop:step]  islice('ABCDEFG', 2, None) --> C D E F G
imap()      func, p, q, ...     func(p0, q0), func(p1, q1), ...             imap(pow, (2,3,10), (5,2,3)) --> 32 9 1000
starmap()   func, seq           func(*seq[0]), func(*seq[1]), ...           starmap(pow, [(2,5), (3,2), (10,3)]) --> 32 9 1000
tee()       it, n               it1, it2 , ... itn splits one iterator into n
takewhile() pred, seq           seq[0], seq[1], until pred fails            takewhile(lambda x: x<5, [1,4,6,4,1]) --> 1 4
izip()      p, q, ...           (p[0], q[0]), (p[1], q[1]), ...             izip('ABCD', 'xy') --> Ax By
izip_longest()  p, q, ...       (p[0], q[0]), (p[1], q[1]), ...             izip_longest('ABCD', 'xy', fillvalue='-') --> Ax By C- D-