CODEWITHSUNDEEP
pythonstatisticspoisson
Jon
Jon

Reputation:

Calculate poisson probability percentage

When you use the POISSON function in Excel (or in OpenOffice Calc), it takes two arguments:

and returns a float.

In Python (I tried RandomArray and NumPy) it returns an array of random poisson numbers. What I really want is the percentage that this event will occur (it is a constant number and the array has every time different numbers - so is it an average?).

for example:

print poisson(2.6,6)

returns [1 3 3 0 1 3] (and every time I run it, it's different).

The number I get from calc/excel is 3.19 (POISSON(6,2.16,0)*100).

Am I using the python's poisson wrong (no pun!) or am I missing something?

Upvotes: 16

Views: 22011

Answers (3)

Ned Batchelder
Ned Batchelder

Reputation: 375804

It is easy to do by hand, but you can overflow doing it that way. You can do the exponent and factorial in a loop to avoid the overflow:

def poisson_probability(actual, mean):
    # naive:   math.exp(-mean) * mean**actual / factorial(actual)

    # iterative, to keep the components from getting too large or small:
    p = math.exp(-mean)
    for i in xrange(actual):
        p *= mean
        p /= i+1
    return p

Upvotes: 14

Alex Coventry
Alex Coventry

Reputation: 70947

scipy has what you want

>>> scipy.stats.distributions
<module 'scipy.stats.distributions' from '/home/coventry/lib/python2.5/site-packages/scipy/stats/distributions.pyc'>
>>> scipy.stats.distributions.poisson.pmf(6, 2.6)
array(0.031867055625524499)

It's worth noting that it's pretty easy to calculate by hand, too.

Upvotes: 26

unwind
unwind

Reputation: 399989

This page explains why you get an array, and the meaning of the numbers in it, at least.

Upvotes: 1

Related Questions