functools管理函数的工具
functools模块提供了一些工具来调整或扩展函数和其他可回调对象,而不必完全重写。
富比较
#!/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() 函数返回一个迭代其,它被索引返回由输入迭代器所选的元素。
islice()与列表的slice操作符参数相同,同样包括开始位置、结束位置和步长,start、step参数是可选的。
示例代码:
#!/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工具,他会重复从输入读到的值,并把他们写到一个命名文件和标准输出。tee()返回的迭代器可以用来为将并行处理的多个算法提供相同的数据集。
#!/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()创建的新迭代器共享其输入迭代器,所以一旦创建了新迭代器,就不应该在使用原迭代器。
转换输入
imap()函数会返回一个迭代器,他对输入迭代器中的值调用一个函数并返回结果。imap()函数的工作方式类似于内置函数map(),只不过要有某个输入迭代器中的元素全部用完。imap()函数都会停止。
示例代码如下:
#!/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()函数与imap类似,但不是由多个迭代器构建一个tuple使用×语法分解一个迭代器中的元素作为映射函数的参数。
示例代码如下:
#!/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()函数返回一个迭代器,能够无限地生成连续整数,地一个数可以作为参数传入(默认为0)