python 自己实现map

python 自己实现map

先来看看map的本来表现

l =[1, 2, 3, 4, 5, 6, 7, 8, 9]
m = map(str,l)
print(next(m))
print(next(m))
print(list(m))

1
2
['3', '4', '5', '6', '7', '8', '9']

map产生的是一个Iterator,是个惰性序列,next(m)会消耗它,可以使用list(m)将整个序列都计算出来

看看智能提示 怎么介绍的

class map(func, *iterables)
map(func, *iterables) --> map object
Make an iterator that computes the function using arguments from each of the iterables.
Stops when the shortest iterable is exhausted.

这是一个可调用的类--->一个map对象,本质是一个Iterator

from collections.abc import Iterator
print(isinstance(m,Iterator))

True

注意到map的参数是 *iterables, 那我多拿传几个Iterable试试

m = map(lambda x,y:x+y,[1,3,5,7],[2,4,8,10,12])
print(list(m))

[3, 7, 13, 17]

map内部应该是 这么调用的: func(1,2) func(3,4) func(5,8) func(7,10) 多余的舍弃

这么将 list1,list2,list3...中的元素按顺序归队呢? zip

list1 = [1,2,3]
list2 = [2,3,4]
list3 = [3,4,5]
z = zip(list1,list2,list3)
print(list(z))  # zip返回的也是一个iterator

[(1, 2, 3), (2, 3, 4), (3, 4, 5)]

def mymap(func, *iterables):
    res = []
    for args in zip(*iterables): ## 相当于zip(list1,list2,list3....)
        res.append(func(*args))  ## 将(1,2,3)=> 1,2,3 解捆绑
    return res

def add(*args):
    sum = 0
    for i in args:
        sum+=i
    return sum

res = mymap(add,list1,list2,list3)
print(res)

[6, 9, 12]

注意到返回的就是一个list,而不是迭代器, 使用yield把每次func调用结果yied出来就成了一个生成器(迭代器)

def mymap(func, *iterables):
    for args in zip(*iterables):
        yield func(*args)  

m = mymap(add,list1,list2,list3)

from collections import Iterator
print(isinstance(m,Iterator))

True

print(next(m))

6

print(list(m))

[9, 12]

当然,这只是一种模拟,cpython内部应该会是用c写的