如何對嵌套列表中的元素求和？

Question

所以，我有一個看起來像這樣的列表

[[[1, 2], [3, 4], [5, 6]],
 [[7, 8], [9, 10], [11, 12]],
 [[13, 14], [15, 16], [17, 18]],
 [[19, 20], [21, 22], [23, 24]]]

我希望它看起來像這樣

[[3, 7, 11],
 [15, 19, 23],
 [27, 31, 35],
 [39, 43, 27]]

即3 = sum([1, 2]), 7 = sum([3, 4]), ... 。

我試過嵌套 for 循環，但我沒有找到任何可以得到想要的結果的東西，有誰知道我該怎么做？

Answer 1

這將很好地完成這項工作，並且 imo 比列表理解更具可讀性。

lists = [[[1, 2], [3, 4], [5, 6]],
        [[7, 8], [9, 10], [11, 12]],
        [[13, 14], [15, 16], [17, 18]],
        [[19, 20], [21, 22], [23, 24]]]

new_lists = []
for nested in lists:
    new_ls = []
    for ls in nested:
        new_ls.append(sum(ls))
    new_lists.append(new_ls)


>>> new_lists
[[3, 7, 11], [15, 19, 23], [27, 31, 35], [39, 43, 47]]

Answer 2

您還可以使用列表推導式：

[[sum(x) for x in triple] for triple in lists]

在上面的列表理解中，三元組將是您的三個雙打列表，因此第一個 for 循環將涵蓋這些。 然后 x 將是三元組內部的每個雙打列表，因此我們對其進行求和，同時通過使用此括號將其保留在原始三元組中：

[sum(x) for x in triple]

輸出：

[[3, 7, 11], [15, 19, 23], [27, 31, 35], [39, 43, 47]]

Answer 3

如果您樂於使用 3rd 方庫，您可以使用 NumPy 並沿單個維度求和：

import numpy as np

A = np.array([[[1, 2], [3, 4], [5, 6]],
              [[7, 8], [9, 10], [11, 12]],
              [[13, 14], [15, 16], [17, 18]],
              [[19, 20], [21, 22], [23, 24]]])

res = A.sum(2)

結果：

array([[ 3,  7, 11],
       [15, 19, 23],
       [27, 31, 35],
       [39, 43, 47]])

另請參閱： 與常規 Python 列表相比，NumPy 的優勢是什么？