python - Python，函数式编程，映射到更高层次

Question

有谁知道如何在 Python 中轻松地将函数映射到嵌套列表中的更高级别，即相当于Map[f, expr, levelspec]Mathematica 中的函数。

Answer 1

您可以轻松滚动自己的

def map_level(f, item, level):
    if level == 0:
        return f(item)
    else:
        return [map_level(f, i, level - 1) for i in item]

>>> double = lambda x: x * 2
>>> data = [[1, 2, 3], [4, 5, 6]]
>>> map_level(double, data, 0)
[[1, 2, 3], [4, 5, 6], [1, 2, 3], [4, 5, 6]]
>>> map_level(double, data, 1)
[[1, 2, 3, 1, 2, 3], [4, 5, 6, 4, 5, 6]]
>>> map_level(double, data, 2)
[[2, 4, 6], [8, 10, 12]]