如何打印/返回 Python 中矩陣中所有值的總和？

Question

我需要創建一個 function，它將矩陣作為參數，然后返回矩陣中所有值的總和。 因此，如果傳遞給 function 的矩陣 = [[12, 4], [9, 6], [5, 7]] 我希望 function 返回值 43

Answer 1

有兩種方法。

(1) 使用 numpy。

L = [[12, 4], [9, 6], [5, 7]]
np.array(L).sum()

(2) 使用迭代。

L = [[12, 4], [9, 6], [5, 7]]
sum([sum(l) for l in L])

Answer 2

Was = [[12, 4], [9, 6], [5, 7]]
>>> count = 0
>>> for i in was:
...     for j in i:
...             count += j
... 
>>> count
43

Answer 3

使用map()

sum(map(sum,[[12, 4], [9, 6], [5, 7]]))

Answer 4

使用itertools.chain.from_iterable()來展平列表，然后調用sum() ：

>>> from itertools import chain
>>> was = [[12, 4], [9, 6], [5, 7]]
>>> sum(chain.from_iterable(was))
43

也可以這樣展平：

>>> sum(number for sublst in was for number in sublst)
43