如何在不使用内置地图或星图的情况下编写类似星图的函数？

Question

这是任务：

创建一个函数my_map_k，该函数采用函数 f 和 k 列出 L ₁ ,...,L _k，对于任意 k ≥ 1，并返回列表 [f(L ₁ [0],...,L _k [ 0]），...，F（L ₁ [N-1]，...，L _ķ [N-1]）]，其中n是最短的长度L_我列表。

暗示。使用 Python 的*符号来处理任意数量的列表作为参数。

例子：

my_map_k(lambda x, y, z: x*y*z, [3, 2, 5], [2, 7, 9], [1, 2])

应该返回[6, 28]。

这是我已经走了多远，但我被卡住了。

def my_map_k(f, *L):
    n = len(min(*L, key=len))
    x=0
    while x < n:        
    return [f(*L[x],) for x in L]
    my_map_k(lambda x, y, z: x*y*z, [3, 2, 5], [2, 7, 9], [1, 2])

问题是，我不能只说有 3 个列表，因为可能还有更多。此外，我看不到如何从所有三个列表中取出第一个元素。

Answer 1

您可以使用依次从每个列表中zip()获取第n个元素，以及一个列表推导式，以使用如此生成的每组参数调用提供的函数：def my_map_k(f, *lists):     return [f(*args) for args in zip(*lists)]这是在行动：>>> my_map_k(lambda x, y, z: x*y*z, [3, 2, 5], [2, 7, 9], [1, 2]) [6, 28]

Answer 2

我想到了：def my_map_k(f, *L):    z = zip(*L)    l = list(z)    return ([f(*x) for x in l])my_map_k(lambda x, y, z: x*y*z, [3, 2, 5], [2, 7, 9], [1, 2])

Answer 3

没有辅助函数的解决方案：from operator import adddef my_map_k(f, *L):    ind = 0    while True:        try:            yield f(*[l[ind] for l in L])        except IndexError:            break        else:            ind += 1result = my_map_k(add, range(5), range(5))print(list(result))# [0, 2, 4, 6, 8]