A fast multivariate hypergeometric calculator with an intuitive language interface.
Find the probability of drawing a target set of objects from a collection, either with or without replacement.
Display the results as a table or a plot in your terminal.
The calculator can be installed via pypi:
pip install urn-calculator
Some calculations may be faster with the gmpy2 library installed. This is optional:
pip install gmpy2
The urn program can be run as a shell:
$ urn
urn>
Computations are described in the following form:
PROBABILITY DRAW [number of things]
FROM [collection]
WHERE [zero or more constraints on draw];
To see the total count of possible draws, replace PROBABILITY with COUNT.
By default, the computation assumes that draws are made without replacement. This can be changed by specifying DRAW [number of things] WITH REPLACEMENT.
Let's look at some examples.
Suppose we want to draw without replacement from an urn containing coloured marbles. We want to see the probability that we see at least 2 red and at most 5 blue (and we don't care about the green marbles):
urn> PROBABILITY DRAW FROM red=5, blue=7, green=3 WHERE red >= 2 AND blue <= 5;
This returns the table of probabilities:
draw size probability
----------- -------------
2 0.0952381
3 0.241758
4 0.406593
5 0.566434
6 0.706294
7 0.818182
8 0.888889
9 0.895105
10 0.818182
11 0.661538
12 0.446154
13 0.2
Note that the query keywords such as FROM and WHERE are not case sensitive. A semicolon ; ends the query. Whitespace is ignored.
We didn't specify a size for our draw, so the program returned all draw sizes with a non-zero probability of meeting our constraints.
By default urn returns float numbers for probabilities. We can make it show exact rational numbers by appending SHOW RATIONAL:
urn> PROBABILITY DRAW 1..5 FROM red=5, blue=7, green=3
WHERE red >= 2 AND blue <= 5
SHOW RATIONAL;
draw size probability
----------- -------------
1 0
2 2/21
3 22/91
4 37/91
5 81/143
Here we also specified a range 1..5 for the draw size. Single draw sizes (e.g. 5) are also permitted.
It's often useful to create a plot (SHOW PLOT) to see the optimal draw size at a glance:
urn> PROBABILITY DRAW FROM red=5, blue=7, green=3
WHERE red >= 2 AND blue <= 5
SHOW PLOT;
┌────────────────────────────────────────────────────────────┐
│ ▝ ▘ │
│ ▘ ▝ │
│ │
│ ▗ │
│ ▘ │
│ │
│ ▘ │
│ │ 0.5
│ ▖ │
│ ▝ │
│ │
│ │
│ ▝ │
│ ▝│
│ │
│▘ │
│▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁│ 0.0
└────────────────────────────────────────────────────────────┘
2 5 7 10 12
To see the same calculation, but in the case where we draw with replacement, we must specify a range and use the WITH REPLACEMENT modifier:
urn> PROBABILITY DRAW 2..13 WITH REPLACEMENT
FROM red=5, blue=7, green=3
WHERE red >= 2 AND blue <= 5
SHOW PLOT;
┌────────────────────────────────────────────────────────────┐
│ ▖ ▝ ▖ │
│ │
│ ▗ ▝ │
│ │
│ ▘ │
│ ▘ │
│ ▖ │ 0.5
│ │
│ ▝ ▗│
│ │
│ │
│ ▝ │
│ │
│ │
│▖ │
│ │
│▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁│ 0.0
└────────────────────────────────────────────────────────────┘
2 5 7 10 12
Finally, we can use OR to specify alternative constraints on our draw.
urn> PROBABILITY DRAW 1..10 FROM red=5, blue=7, green=3
WHERE red >= 2 AND blue <= 3
OR blue > 0 AND green > 1
OR blue = 2 AND red >= 2 AND green <= 2
SHOW PLOT;
┌────────────────────────────────────────────────────────────┐
│ ▗ ▝ ▘ ▝│ 1.0
│ │
│ ▘ │
│ ▖ │
│ ▗ │
│ ▖ │
│ │
│ │
│ │ 0.5
│ │
│ │
│ ▗ │
│ │
│ │
│ │
│ │
│▖▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁│ 0.0
└────────────────────────────────────────────────────────────┘
2 4 6 8 10
To exit the shell, type quit:
urn> quit;
Exiting urn shell.