functools管理函數(shù)的工具
functools模塊提供了一些工具來調(diào)整或擴(kuò)展函數(shù)和其他可回調(diào)對(duì)象,而不必完全重寫。
富比較
#!/usr/bin/env python
import functools
import inspect
from pprint import pprint
@functools.total_ordering
class Myobject(object):
def __init__(self,val):
self.val = val
def __eq__(self, other):
print ' eq------>testing __eq__(%s, %s)' % (self.val, other.val)
return self.val == other.val
def __gt__(self, other):
print ' gt------->testing __gt__(%s, %s) % (self.val, other.val)'
return self.val >other.val
print '.Methods:\n'
pprint(inspect.getmembers(Myobject,inspect.ismethod))
a = Myobject(1)
b = Myobject(2)
print '\nComparisons:'
for expr in ['a<b','a<=b','a==b','a>=b','a>b','']:
print '\n%-6s:' % expr
result = eval(expr)
print 'result of %s: %s' % (expr, result)
合并分解迭代器
from itertools import *
for i in chain([1,2,3],['a','b','c']):
print i
print
print '-'*15,'izip()', '-'*15
for j in izip([1,2,3],['a','b','c']):
print j
islice()
islice() 函數(shù)返回一個(gè)迭代其,它被索引返回由輸入迭代器所選的元素。
islice()與列表的slice操作符參數(shù)相同,同樣包括開始位置、結(jié)束位置和步長(zhǎng),start、step參數(shù)是可選的。
示例代碼:
#!/usr/bin/env python
from itertools import *
print 'stop at 5:'
for i in islice(count(), 5):
print i,
print '\n'
print 'start at 5, stop at 10:'
for i in islice(count(), 5, 10):
print i,
print '\n'
print 'by tens to 100:'
for i in islice(count(),0,100,10):
print i,
print '\n'
tee
tee()的語義類似于unix tee工具,他會(huì)重復(fù)從輸入讀到的值,并把他們寫到一個(gè)命名文件和標(biāo)準(zhǔn)輸出。tee()返回的迭代器可以用來為將并行處理的多個(gè)算法提供相同的數(shù)據(jù)集。
#!/usr/bin/env python
from itertools import *
r = islice(count(),5)
print r
i1, i2 = tee(r)
print 'i1:', list(i1)
print 'i2:', list(i2)
tee()創(chuàng)建的新迭代器共享其輸入迭代器,所以一旦創(chuàng)建了新迭代器,就不應(yīng)該在使用原迭代器。
轉(zhuǎn)換輸入
imap()函數(shù)會(huì)返回一個(gè)迭代器,他對(duì)輸入迭代器中的值調(diào)用一個(gè)函數(shù)并返回結(jié)果。imap()函數(shù)的工作方式類似于內(nèi)置函數(shù)map(),只不過要有某個(gè)輸入迭代器中的元素全部用完。imap()函數(shù)都會(huì)停止。
示例代碼如下:
#!/usr/bin/env python
from itertools import *
print 'Doubles:'
for i in imap(lambda x:2*x, xrange(5)):
print i
print 'multiples:'
for i in imap(lambda x, y:(x,y, x*y), xrange(5),xrange(5,10)):
print '%d * %d = %d' % i
starmap()函數(shù)與imap類似,但不是由多個(gè)迭代器構(gòu)建一個(gè)tuple使用×語法分解一個(gè)迭代器中的元素作為映射函數(shù)的參數(shù)。
示例代碼如下:
#!/usr/bin/env python
from itertools import *
values = [(0,5),(1,6),(2,7),(3,8),(4,9)]
for i in starmap(lambda x,y:(x, y, x*y), values):
print '%d * %d = %d' % i
生成新值
count()函數(shù)返回一個(gè)迭代器,能夠無限地生成連續(xù)整數(shù),地一個(gè)數(shù)可以作為參數(shù)傳入(默認(rèn)為0)