Python-сообщество

I159 · Апрель 14, 2011 13:01:06

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

def add_coll(*args):
    for i in args:
        if isinstance(i, tuple) or isinstance(i, list):
            add_coll(*i)
    return sum(args)

По задумке, функция должна воспроизводить себя до того момента, пока вложенные объекты будут итерируемые. А в итоге сложить все элементы полученного списка. При введении списка

>>> add_coll([1,2,3])

написанная функция выдает ошибку:

TypeError: unsupported operand type(s) for +: ‘int’ and ‘list’

Вместо

return sum(args)

, подставил

print args

и получил:

>>> (1, 2, 3)
>>> ([1, 2, 3],)

Мне не понятен механизм, как список заново попадает в аргументы? И как сделать, чтобы функция работала, в соответсвии с задачей?
Спасибо.

sp3 · Апрель 14, 2011 13:33:25

def add_coll(*args):
    m = 0
    for i in args:
        if isinstance(i, tuple) or isinstance(i, list):
            m += add_coll(*i)
        else:
             m += i
    return m

print add_coll([1,2,3,[[1,2],2],(1,2)])

Python-сообщество

Уведомления

#1 Апрель 14, 2011 13:01:06

Рекурсия

#2 Апрель 14, 2011 13:33:25

Рекурсия

Board footer