算法工具

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','']:
    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)

你可能感兴趣的:(算法工具)