More efficient way to sum values in multiple arrays

Question

Hello is there better way to sum multiple arrays? Because what I did is nested loop and i am not fan of this solution.

a = [1,2,8,7,4], [6,7,2,1,6], [7,5,1,2,3]
sum = []
total = 0

for i in a:
    for j in i: 
        total = total + j
        sum.append(total)

print(sum[-1])

expected output: 62

from stackoverflow.com/a/952946/16668765 you can try sum(sum(a, [])) oh it's super inefficient but fun — folen gateis
– folen gateis, Commented Aug 21, 2021 at 22:47

ThePyGuy · Accepted Answer · 2021-08-21 22:38:46Z

1

You can just use sum inside the comprehension (generator expression) for each sub-lists in the list, then pass it to the sum builtin again.

>>> sum(sum(i) for i in a)
62

Or, map each sub-list to the sum, and pass it to sum

>>> sum(map(sum, a))
62

edited Aug 21, 2021 at 22:38

answered Aug 21, 2021 at 22:34

ThePyGuy

18.5k5 gold badges24 silver badges55 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

ThePyGuy Over a year ago

Yeah, map also does the same; however, I'm fond of the comprehension.

ThePyGuy Over a year ago

Yes, it's comprehension but not the list comprehension, it's generator expression, and people also refer to it as generator comprehension.

Collectives™ on Stack Overflow

More efficient way to sum values in multiple arrays

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related