python element wise add in list of list with other list

Question

I checked similar problem at sum of N lists element-wise python

but mine is a little bit more complicate cause I am adding list of list with another list of list

my list looks like it below

[[0         0.014285714 0.035600016]
,[0.014285714   0           0.038359389]
,[0.035600016   0.038359389 0]]


[[0         0.014285714 0.035600016]
,[0.014285714   0           0.038359389]
,[0.035600016   0.038359389 0]]

so adding these two then results has 3 by 3 matrix

how could I efficiently add these two lists?

James Sapam · Accepted Answer · 2014-04-06 14:38:26Z

3

Can i solve using numpy:

>>> a = [[1, 2, 3], [1, 2, 3], [1, 2, 3]]
>>> b = [[4, 5, 6], [4, 5, 6], [4, 5, 6]]

>>> from numpy import array, sum

>>> list(array(a) + array(b))
[array([5, 7, 9]), array([5, 7, 9]), array([5, 7, 9])]

OR

>>> sum([a, b], axis=0)
 array([[5, 7, 9],
       [5, 7, 9],
       [5, 7, 9]])
>>>

answered Apr 6, 2014 at 14:38

James Sapam

17.2k12 gold badges55 silver badges77 bronze badges

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

Comments

anon582847382 · Accepted Answer · 2014-04-06 14:35:30Z

3

You tagged your question numpy, so I assume this is OK:

import numpy as np

a = np.matrix(a)
b = np.matrix(b)
c = a + b

Where a and b are your two lists, and c is the two added together.

By far the simplest with numpy. :)

answered Apr 6, 2014 at 14:35

anon582847382

20.5k5 gold badges58 silver badges60 bronze badges

1 Comment

Daniel Over a year ago

It is advisable to avoid the matrix class unless you specifically want the caveats of the matrix class.

sshashank124 · Accepted Answer · 2014-04-06 14:38:14Z

3

If you are working with numpy matrices, @AlexThornton has the best answer

If you are dealing with lists, you should do:

summed_list = [[c+d for c,d in zip(i,j)] for i,j in zip(a,b)]   #a,b are your two lists

Simple example:

>>> a = [[1,2,3],[1,2,3],[1,2,3]]
>>> b = [[4,5,6],[4,5,6],[4,5,6]]

>>> summed_list = [[c+d for c,d in zip(i,j)] for i,j in zip(a,b)]   #a,b are your two lists

>>> print summed_list
[[5,7,9],
 [5,7,9],
 [5,7,9]]

edited Apr 6, 2014 at 14:38

answered Apr 6, 2014 at 14:29

sshashank124

32.3k10 gold badges72 silver badges76 bronze badges

Comments

Pratik K. · Accepted Answer · 2017-06-15 22:19:36Z

Here is a recursive solution using 'vanilla' python and recursion (This solution works for nested tuples or lists)

from operator import add
def list_recur(l1, l2, op = operator.add):
    if not l1:
        return type(l1)([])
    elif isinstance(l1[0], type(l1)):
        return type(l1)([list_recur(l1[0], l2[0], op)]) + \
list_recur(l1[1:],l2[1:], op)
    else:
        return type(l1)([op(l1[0], l2[0])]) + \
list_recur(l1[1:], l2[1:], op)

This solution takes arguments list1, list2 and optionally the operation you want to perform element-wise (by default addition).

For example, list_recur([ [1, 2, 3, [4, 5] ], [1, 2, 3, [4, 5] ]) will return [ [2, 4, 6, [8, 10] ].

More complex examples will work too: list_recur([[[[[1,[[[[2, 3]]]]]]]], [[[[3, [[[[4, 1]]]]]]]], lambda x, y: \ x < y) will return [[[[True, [[[[True, False]]]]]]]] `

Collectives™ on Stack Overflow

python element wise add in list of list with other list

4 Answers 4

Comments

1 Comment

Simple example:

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

1 Comment

Simple example:

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related