algorithm - 总结列表和列表列表的 Pythonic 方式

Question

我正在尝试找到一种在同一函数中对列表和列表列表求和的巧妙方法，到目前为止我已经得到:

import operator
"""
 Fails late for item = ['a', 'b']
"""   
def validate(item):
    try:
        return sum(item) == sum(range(1, 10))
    except TypeError:
        return sum(reduce(operator.add, item)) == sum(range(1, 10))

"""
 Not valid for item = [1,2,[3,4,5]]
"""
def validate2(item):
        if isinstance(item[0], int):
            return sum(item) == sum(range(1, 10))
        else:
            return sum(reduce(operator.add, item)) == sum(range(1, 10))


print validate([1, 2, 3, 4, 5, 6, 7, 8, 9])
print validate([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

print validate2([1, 2, 3, 4, 5, 6, 7, 8, 9])
print validate2([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

...但这些对我来说似乎都不正确(文档字符串中的原因)。我想知道的是，是否有更好的方法来汇总列表和列表的列表，而不需要我捕获异常或在函数决定要做什么之前实际分析列表。

显然，我仍然希望 ['a', 'b'] 无效。

Answer 1

也许您会发现先将列表展平更容易？

def flatten(xs):
     for x in xs:
        try:
            sub = iter(x)
        except TypeError:
            yield x
        else:
            for y in flatten(sub):
                yield y

有了上面的内容，你可以这样做:

In [4]: fs = flatten([1,2,[3,4,[5,6],7],8,[9,10]])

In [5]: list(fs)
Out[5]: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]