Блог

Мне нужно перебирать вложенные списки в python, где элементы списка являются вызовами функций (первый элемент в списке-функция, все остальные элементы являются аргументами этой функции). Эти аргументы также сами по себе могут быть функциями с аргументами, отсюда и вложенность. Список также может содержать одну функцию без аргументов. Вот так: ['foo','arg1','arg2'] Или: [['foo','arg1','arg2'],['bar'],['baz','arg1']] Или: ['foo','arg1',['baz','arg1']] И т. Д.

Вот пример использования фиктивных функций суммы и продукта… Сумма складывается, произведение умножается; Если бы мы оценили функции, то получили бы 10.

[«сумма»,[«продукт»,2,3],4]

Я могу просто передать функцию и аргументы другой функции, которая выполняет фактический «вызов» моих безопасных функций, но у меня возникают проблемы с рекурсивным повторением списка. Вот печальная попытка:

 def process_list(l):
        if isinstance(l,list):
            results = []
            for item in l:
                if isinstance(item,list):
                    func = item[0]
                    args = item[1:]
                    for idx, arg in enumerate(args):
                        if isinstance(arg,list):
                            args[idx] = process_list(arg)
                    results.append(function_caller_result(func,args)) #what if that's a list?
                else:
                    results.append(item)
            return results
        return l


 

Любая помощь здесь будет очень признательна! Заранее спасибо! 🙂

Вы можете попробовать эту рекурсивную функцию, которая работает для некоторых тестов, в которых решение Ajax1234 не работает.

Я заметил, что их ввод имеет форму [['sum',['product',2,3],4]] , а ['sum',['product',2,3],4] не показан в вашем вопросе.

Это решение использует ваш список входных ['sum',['product',2,3],4] данных вместе с другими тестами.

 from operator import add, sub, mul, truediv

funcs = {'sum': add, 'minus': sub, 'product': mul, 'divide': truediv}  # you can use your functions instead of this

def process(seq):
    if isinstance(seq, list):
        return funcs[seq[0]](*(process(i) for i in seq[1:]))
    return seq

print(process(['sum',['product',2,3],4]))
print(process(['sum',['product',['minus', 2, 3],3],4]))
print(process(['sum',['product',['minus', 2, 3],3],['divide', 3, 4]]))
 

Выход

 10
1
-2.25
 

Если вы хотите один лайнер, то

 def process(seq):
    return seq if not isinstance(seq, list) else funcs[seq[0]](*map(process, seq[1:]))
 

Редактировать:

Это не сработает в случае, который я объяснил в комментарии, решение Ajax1234 решает эту проблему и объясняет причину изменения списка ввода и логики.

Вы пытаетесь создать своего рода рекурсивную функцию, но вручную. Это не сработает, если вложенность больше, чем вы закодировали.

Вместо этого напишите рекурсивную функцию:

 my_list = ['sum',['product',2,3],4]

def process(l):
    l = l.copy()
    for item in l:
        if isinstance(item,list):
            l.replace(item, process(item))
    return function_caller_result(*l) 
    # suppose l = [1, [2], '3']
    # func(*l) => func(1, [2], '3')

result = process(my_list)
 

Вы можете использовать рекурсию при распаковке:

 eval_d = {'sum':lambda x, y:x y, 'product':lambda x, y:x*y} #dict to store function names with an associated callable
data = [['sum',['product',2,3],4]]
def eval_f(d):
   r = [eval_d[a](*[i if not isinstance(i, list) else eval_f([i]) for i in b]) 
        for a, *b in d]
   #r now stores the result from the function calls
   return sum(r) #for this example to work, a cumulate sum is returned, but you can also just return the list

print(eval_f(data))
 

Выход:

 10
 

Цель этого решения-работать в сценариях, в которых eval_f передается список списков с сигнатурами функций [['foo','arg1','arg2'],['bar'],['baz','arg1']] , т. е. Таким образом, отдельные вызовы функций должны быть заключены в список, чтобы гарантировать, что шаблон будет сохранен: ['foo','arg1','arg2'] будет передан eval_f и [['foo','arg1','arg2']] .