Как бы я суммировал многомерный массив в самом лаконичном питоне?

Question

Ближайшим был этот столбец суммирования .

Так что я сделаю нечто подобное в своем вопросе:

Скажем, у меня есть список Python 2D, как показано ниже:

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

Я могу получить итоговые значения строки с пониманием списка:

row_totals = [ sum(x) for x in my_list ]

В одной строке, как я могу сложить весь 2d-массив?

27

Answer 1

Вы можете сделать так же просто, как

sum(map(sum, my_list))

или, альтернативно,

sum(sum(x) for x in my_list))

и назвать это днем, если вы не ожидаете более двух измерений.Обратите внимание, что первое решение, скорее всего, не самое быстрое (как во время выполнения) решение из-за использования map().Проведите сравнительный анализ и при необходимости сравните.

Наконец, если вы обнаружите, что используете многомерные массивы, рассмотрите возможность использования NumPy и его превосходных дружественных к массиву функций.Вот краткий отрывок для вашей проблемы:

import numpy as np

my_list = np.array([[1,2,3,4], [2,4,5,6]])
np.sum(my_list)

Это подойдет для любого числа измерений, которые могут иметь ваши массивы.

Answer 2

Другое решение с использованием itertools:

>>> from itertools import chain
>>> my_list = [ [1,2,3,4], [2,4,5,6] ]
>>> sum(chain(*my_list))
27

Answer 3

>>> sum ( [ sum(x) for x in [[1,2,3,4], [2,4,5,6]] ] )
27

Answer 4

>>> from itertools import chain
>>> my_list = [[1,2,3,4], [2,4,5,6]]
>>> sum(chain.from_iterable(my_list))
27

Answer 5

Вы можете использовать сумму, чтобы сначала сложить внутренние списки, а затем суммировать полученный список:

>>> my_list = [ [1,2,3,4], [2,4,5,6] ]

>>> sum(my_list, [])
[1, 2, 3, 4, 2, 4, 5, 6]

>>> sum(sum(my_list, []))
27