python - 如何将函数应用于列表中的列表

Question

我有一个列表和一个函数:

t = [3, [1], [2, [1], [1]]]

f = lambda x: x**2

我想要这样的结果:

[9, [1], [4, [1], [1]]]

我尝试使用 map 功能，但它似乎不起作用

执行此操作时出现类型错误:

list(map(f, t))

Answer 1

如果t的元素是另一个列表，需要映射函数，而不是调用函数。如果您希望它处理任意级别的嵌套，f 需要递归。

t = [3, [1], [2, [1], [1]]]

def f(x):
    if isinstance(x, list):
        return map(f, x)
    else:
        return x**2

print(map(f, t))