首先,为Python程序员展示了一个通用的反模式,不要循环索引,循环对象本身。E、 g组:for item in b:
do_something(item)
而不是:
^{pr2}$
它更清晰、更简单、更快。在
也就是说,你遇到的主要问题是其中一项不是列表,所以它没有长度。在
这里一个更好的选择是用生成器表达式和^{}展开列表,然后使用the ^{} builtin function对元素求和。在>>> import collections
>>> import itertools
>>> b = [[1,2], [3,4], [5,6], 1]
>>> list(itertools.chain.from_iterable(item if isinstance(item, collections.Iterable) else [item] for item in b))
[1, 2, 3, 4, 5, 6, 1]
>>> sum(itertools.chain.from_iterable(item if isinstance(item, collections.Iterable) else [item] for item in b))
22
我们需要生成器表达式,因为itertools.chain()不能处理不可编辑的项,因此我们将任何表达式放入一个列表中以解决这一问题。在
另一种方法是制造自己的发电机:def chain_mixed(iterable):
for item in iterable:
try:
for subitem in item:
yield subitem
except TypeError:
yield item
然后你可以简单地做:sum(chain_mixed(b))