執行映射返回列表而不是列表

Question

我想實現我自己的Python映射函數版本。我稱它爲my_map。約束條件是使用函數式編程（如果語句允許，循環不允許）。

這裏的實現：

# f is the function it takes in, and ls is the list to apply this function over. 
def my_map(f, ls): 
    if len(ls) != 1: 
     ret = my_map(f, ls[1:]) 
     return [f(ls[0])] + [ret] 
    else: 
     return f(ls[0]) 

但是當我運行這個具有以下輸入：

def f(x): 
    return x * x 

my_map(f, [1,2,3,4])回報[1, [4, [9, 16]]]

而map(f, [1,2,3,4])回報[1, 4, 9, 16]，這是我的期望。

關於如何獲得預期結果的任何想法？

Answer 1

return [f(ls[0])] + [ret] 

應該

return [f(ls[0])] + ret 

。您正在從ret創建一個新的單元素列表，但只有第一個應該是單元素列表。

此外，發電機可能會更有效。你正在製作大量的部分列表。

def my_map(f, ls): 
    it = iter(ls) 

    def go(): 
     yield f(next(it)) 
     yield from go() 

    return list(go())