对多级列表中的每个项目执行操作

时间:2013-04-04 09:48:29

标签: python list

假设我有一个列表:

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

如何以Pythonic方式对列表中的所有元素应用函数(例如int)?
例如:

>>> func_on_all(li, int)
[0, [1, 2], [3, [4, 5], 6, [7], 8], 9]

4 个答案:

答案 0 :(得分:3)

您可以使用此功能:

def to_int(lst):
    for i in lst:
        if isinstance(i, list):
            yield list(to_int(i))
        else:
            yield int(i)

这是一个测试运行:

>>> li = ['0', ['1', '2'], ['3', ['4', '5'], '6', ['7'], '8'], '9']
>>> list(to_int(li))
[0, [1, 2], [3, [4, 5], 6, [7], 8], 9]

请注意,这仅适用于嵌套列表,并且只能执行int功能。我们可以使函数更加通用:

def deep_map(lst, f=int):  # the parameter `f` is the function
    for i in lst:
        if isinstance(i, collections.Iterable) and not isinstance(i, str):
            yield list(deep_map(i, f))
        else:
            yield f(i)

这将允许我们使用除列表之外的嵌套集合(例如元组),并且我们可以指定我们自己的函数来调用列表。

答案 1 :(得分:2)

第一个问题:

f = lambda obj, el_func: [f(el, el_func) for el in obj] if isinstance(obj, list) else el_func(obj)
f(li, int)

答案 2 :(得分:1)

我喜欢我的版本:)。

def recursive_map(func, iterable, kind=list):
    if isinstance(iterable, kind):
        return map(lambda x: recursive_map(func, x), iterable)
    else:
        return func(iterable)

将您希望该函数使用的类传递到kind

感谢Volatility,忘了iter:)

答案 3 :(得分:0)

这是另一个:

def deploy(l, out) :

    if isinstance(l, list) :
        for i in l :
            deploy(i, out)
    else :
        out.append(l)


def applyFuncOnMultilevelList(li, func) :
    out = []
    deploy(li, out) 
    for i in out :
        func(i)

li = ['0', ['1', '2'], ['3', ['4', '5'], '6', ['7'], '8'], '9']
applyFuncOnMultilevelList(li, print)