How to divide list item by list item from another list using Python?

165 views Asked by At

I would like to divide list items inside two lists.

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

How can I divide a by b to obtain this output:

c = [[0.2, 0, 0.5], [0, 0, 0], [0.333], [0.333]]

Can anyone help me?

3

There are 3 answers

0
Ami Tavory On BEST ANSWER

Using itertools.izip (Python2.7):

import itertools

[[float(aaa) / bbb for (aaa, bbb) in itertools.izip(aa, bb)] \
    for (aa, bb) in itertools.izip(a, b)]
2
Bakuriu On

Zip the two lists and use a list comprehension:

from __future__ import division   # in python2 only
result = [[x/y for x,y in zip(xs, ys)] for xs, ys in zip(a, b)]

Sample run:

In [1]: a = [[1, 0, 2], [0, 0, 0], [1], [1]]
   ...: b = [[5, 6, 4], [6, 6, 6], [3], [3]]
   ...: 

In [2]: result = [[x/y for x,y in zip(xs, ys)] for xs, ys in zip(a, b)]

In [3]: result
Out[3]: [[0.2, 0.0, 0.5], [0.0, 0.0, 0.0], [0.3333333333333333], [0.3333333333333333]]
0
John La Rooy On
from __future__ import division # for Python2

[[x / y for x, y  in zip(*item)] for item in zip(a, b)]