Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How do I divide the members of a list by the corresponding members of another list in Python?

Tags:

python

Let's say I have two data sets. I have a week-by-week tally of users who tried my service.

trials = [2,2,2,8,8,4]

And I have a week by week tally of trial users who signed up.

conversions = [1,0,2,4,8,3]

I can do it pretty quickly this way:

conversion_rate = []
for n in range(len(trials)):
   conversion_rate.append(conversions[n]/trials[n])

Can you think of a more elegant way?

Bonus: The result of this is a list of ints [0, 0, 1, 0, 1, 0] , not a list of floats. What's the easiest way to get a list of floats?

like image 470
Rich Armstrong Avatar asked Sep 16 '10 22:09

Rich Armstrong


2 Answers

Use zip:

[c/t for c,t in zip(conversions, trials)]

The most elegant way to get floats is to upgrade to Python 3.x.

If you need to use Python 2.x then you could write this:

>>> [float(c)/t for c,t in zip(conversions, trials)]
[0.5, 0.0, 1.0, 0.5, 1.0, 0.75]

Alternatively you could add this at the start of your file:

from __future__ import division

But note that this will affect the behaviour of division everywhere in your program, so you need to check carefully to see if there are any places where you want integer division and if so write // instead of /.

like image 177
Mark Byers Avatar answered Oct 07 '22 16:10

Mark Byers


How about

trials = [2,2,2,8,8,4]
conversions = [1,0,2,4,8,3]
conversion_rate = [conversion / trials[n] for n, conversion in enumerate(conversions)]
like image 42
Day Avatar answered Oct 07 '22 16:10

Day