在列表中使用sum（）時出錯

Question

我有一個list_of_lists，我想創建一個包含所有內部列表的所有元素的列表。我不明白爲什麼不能使用sum來實現這一點。

list_of_lists = [['element 1','element 2'],['element 1']] 
print(sum(list_of_lists)) 
>> TypeError: unsupported operand type(s) for +: 'int' and 'list' 

如果我使用reduce正常工作：

from functools import reduce 
print(reduce(lambda x,y: x+y, list_of_lists)) 
>> ['element 1', 'element 2', 'element 1'] 

Answer 1

因爲sum有一個默認的0初始化一個空列表，而不是初始化：

print(sum(list_of_lists, [])) 
# ['element 1', 'element 2', 'element 1'] 

但是，它不是一個好想法與sum（可怕的表現）平展名單，你可以使用嵌套綜合on或itertools.chain.from_iterable這是爲此創建的。

Answer 2

我要闡述@ moses-koledoe的答案。爲了扁平化這些列表（有效地，這就是你想要實現的），僱用itertools.chain。

from itertools import chain  
list_of_lists = [['element 1','element 2'],['element 1']] 
list(chain(*list_of_lists)) 
>> ['element 1', 'element 2', 'element 1'] 

如果你想避免拆包語法（*list_of_lists），使用相同的功能的from_iterable變化：

list(chain.from_iterable(list_of_lists)) 
>> ['element 1', 'element 2', 'element 1'] 

事實上，第二種方法可能在因爲大名單一點點更好的性能它不會構造調用chain函數所需的參數的中間元組，但可直接在提供的迭代器（可能是一個生成器）上工作。