總和列表的列表簡潔特定元素

Question

我有一個變量堆棧：

stacks = [[1, 2, 3], [[4, 5, 6], [1, 2, 3]]] 

從這個我想創造的高度，其中每個高度的元素在每個指標1和的另一個列表堆疊。在上面的例子中，高度將是：

heights = [2, 7] 

凡2是疊層[0] [1]和圖7是堆疊[1] [0] [1] +堆疊[1] [1] [1] 。對不起，如果它之前還不清楚。我如何使用列表解析，地圖和/或簡化來簡潔地做到這一點？

Answer 1

假設stacks是完全按照你所描述：

>>> stacks = [[1, 2, 3], [[4, 5, 6], [1, 2, 3]]] 
>>> wrapped = (s if isinstance(s[0], list) else [s] for s in stacks) 
>>> total = [sum(x[1] for x in w) for w in wrapped] 
>>> total 
[2, 7] 

這將是更自然的，恕我直言，如果stacks元素總是列表的列表：

>>> stacks = [[[1, 2, 3]], [[4, 5, 6], [1, 2, 3]]] 
>>> total = [sum(x[1] for x in w) for w in stacks] 
>>> total 
[2, 7]