Come potrei sommare un array multidimensionale nel pitone più succinto?

Question

Il più vicino era questo summing columns.

quindi dovrò fare qualcosa di simile nella mia interrogazione:

dire che ho una lista Python 2D, come di seguito:

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

posso ottenere i totali di riga con una lista di comprensione:

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

In una riga, come posso sommare l'intero array 2d?

27 

Answer 1

Si può fare facile come

sum(map(sum, my_list)) 

o in alternativa

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

e chiamare un giorno, se non ci si aspetta più di 2 dimensioni. Si noti che la prima soluzione non è probabilmente la soluzione più veloce (come nel tempo di esecuzione), a causa dell'uso di map(). Confronta e confronta secondo necessità.

Infine, se ci si trova a utilizzare matrici multidimensionali, considerare l'utilizzo di NumPy e le sue funzioni di array superiori. Ecco un breve estratto del tuo problema:

import numpy as np 

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

Questo funzionerebbe per qualsiasi numero di dimensioni che i tuoi array potrebbero avere.

Answer 2

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

Answer 3

Un'altra soluzione utilizzando itertools:

>>> from itertools import chain 
>>> my_list = [ [1,2,3,4], [2,4,5,6] ] 
>>> sum(chain(*my_list)) 
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

È possibile utilizzare somma per aggiungere prima le liste interne insieme e poi sommare la lista appiattita risultante:

>>> 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