簡介
官方描述:Functional tools for creating and using iterators.即用於建立高效迭代器的函式。
itertools.chain(*iterable)
將多個序列作為一個單獨的序列返回。 例如:
1 2 3 |
import itertools for each in itertools.chain('i', 'love', 'python'): print each |
輸出:
1 2 3 4 5 6 7 8 9 10 11 |
i l o v e p y t h o n |
itertools.combinations(iterable, r)
返回指定長度的”組合” 例如:
1 2 3 |
import itertools for each in itertools.combinations('abc', 2): print each |
輸出:
1 2 3 |
('a', 'b') ('a', 'c') ('b', 'c') |
itertools.combinations_with_replacement(iterable, r)
返回指定長度的“組合”,組合內元素可重複 例如:
1 2 3 |
import itertools for each in itertools.combinations_with_replacement('abc', 2): print each |
輸出:
1 2 3 4 5 6 |
('a', 'a') ('a', 'b') ('a', 'c') ('b', 'b') ('b', 'c') ('c', 'c') |
itertools.product(*iterable[,repeat])
返回指定長度的所有組合,可理解為笛卡爾乘積 例如:
1 2 3 |
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的排列 例如:
1 2 3 |
import itertools for value in itertools.permutations('abc', 2): print value |
輸出:
1 2 3 4 5 6 |
('a', 'b') ('a', 'c') ('b', 'a') ('b', 'c') ('c', 'a') ('c', 'b') |
itertools.compress(data,selector)
返回selector為True的data對應元素 例如:
1 2 3 |
import itertools for each in itertools.compress('abcd', [1, 0, 1, 0]): print each |
輸出:
1 2 |
a c |
itertools.count(start=0,step=1)
返回以start開始,step遞增的序列,無限遞增 例如:
1 2 3 |
import itertools for each in itertools.count(start=0, step=2): print each |
輸出:
1 2 3 4 5 |
1 2 3 . . |
itertools.cycle(iterable)將迭代器進行無限迭代 例如:
1 2 3 |
import itertools for each in itertools.cycle('ab'): print each |
輸出:
1 2 3 4 5 6 |
a b a b . . |
itertools.dropwhile(predicate, iterable)
直到predicate為真,就返回iterable後續資料, 否則drop掉 例如:
1 2 3 |
import itertools for each in itertools.dropwhile(lambda x: x<5, [2,1,6,8,2,1]): print each |
輸出:
1 2 3 4 |
6 8 2 1 |
itertools.groupby(iterable[,key])
返回一組(key,itera),key為iterable的值,itera為等於key的所有項 例如:
1 2 3 |
import itertools for key, vale in itertools.groupby('aabbbc'): print key, list(vale) |
輸出:
1 2 3 |
a ['a', 'a'] b ['b', 'b', 'b'] c ['c'] |
itertools.ifilter(predicate, iterable)
返回predicate結果為True的元素迭代器,如果predicate為None,則返回所有iterable中為True的項 例如:
1 2 3 |
import itertools for value in itertools.ifilter(lambda x: x % 2, range(10)): print value |
輸出:
1 2 3 4 5 |
1 3 5 7 9 |
itertools.ifilterfasle(predicate,iterable)
返回predicate為False的元素,如果predicate為None,則返回所有iterable中為False的項 例如:
1 2 3 |
import itertools for value in itertools.ifilterfalse(lambda x: x % 2, range(10)): print value |
輸出:
1 2 3 4 5 |
0 2 4 6 8 |
itertools.imap(function,*iterables)
相當於迭代器方式的map() 例如:
1 2 3 |
import itertools for value in itertools.imap(lambda x, y: x+y, (1,2,3), (4,5,6)): print value |
輸出:
1 2 3 |
5 7 9 |
itertools.islice(iterable, start,stop[,step])
相當於迭代器方式的切片操作 例如:
1 2 3 |
import itertools for value in itertools.islice('abcdefg', 1, 4, 2): print value |
輸出:
1 2 |
b d |
itertools.repeat(object,[,times])
不停的返回object物件,如果指定了times,則返回times次 例如:
1 2 3 |
import itertools for value in itertools.repeat('a', 2): print value |
輸出:
1 2 |
a a |
itertools.starmap(function,iterable)
返回function(iter)的值,iter為iterable的元素 例如:
1 2 3 |
import itertools for value in itertools.starmap(lambda x, y: x * y, [(1, 2), (3, 4)]): print value |
輸出:
1 2 |
2 12 |
itertools.takewhile(predicate,iterable)
如果predicate為真,則返回iterable元素,如果為假則不再返回,break. 例如:
1 2 3 |
import itertools for value in itertools.takewhile(lambda x: x < 5, [1, 3, 5, 6]): print value |
輸出:
1 2 |
1 3 |