自己实现Python中的map函数

'''
自定义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