'''
自定义map
'''
def map(func,*iterators):
try:
i = 0
while 1:
yield func(*[j[i] for j in iterators])
i+=1
except IndexError :
pass
这里使用yield生成器实现map函数,功能基本和map一致。
python内置的map是一个类,那么我们也可以实现map类
class map:
def __init__(self,func,*args):
self.iterators=args
self.func=func
def __iter__(self):
return self.generator()
def generator(self):
iterators,func=self.iterators,self.func
try:
i = 0
while 1:
yield func(*[j[i] for j in iterators])
i += 1
except IndexError:
pass
测试代码:
f=lambda x,y,z,w:x==y
a=[1,2,44,4,5,6,7,8,9]
b=[1,2,3,4,11,6,7,8,9]
print(list(map(f,a,b,a,b)))
测试结果:
[True, True, False, True, False, True, True, True, True]
Process finished with exit code 0