In probability and statistics, a probability mass function (PMF) assigns the probability that a discrete random variable takes a specific value and is a key tool for defining a discrete probability distribution. Unlike a continuous probability density function (PDF), which relates to continuous variables and requires integration over intervals to find probabilities, the PMF directly gives probability values for discrete outcomes. These functions apply to both scalar and multivariate random variables with discrete domains. The random variable value with the highest probability mass is known as the mode.
Formal definition
Probability mass function is the probability distribution of a discrete random variable, and provides the possible values and their associated probabilities. It is the function p : R → [ 0 , 1 ] {\displaystyle p:\mathbb {R} \to [0,1]} defined by
p X ( x ) = P ( X = x ) {\displaystyle p_{X}(x)=P(X=x)}
for − ∞ < x < ∞ {\displaystyle -\infty <x<\infty } ,4 where P {\displaystyle P} is a probability measure. p X ( x ) {\displaystyle p_{X}(x)} can also be simplified as p ( x ) {\displaystyle p(x)} .5
The probabilities associated with all (hypothetical) values must be non-negative and sum up to 1,
∑ x p X ( x ) = 1 {\displaystyle \sum _{x}p_{X}(x)=1} and p X ( x ) ≥ 0. {\displaystyle p_{X}(x)\geq 0.}
Thinking of probability as mass helps to avoid mistakes since the physical mass is conserved as is the total probability for all hypothetical outcomes x {\displaystyle x} .
Measure theoretic formulation
A probability mass function of a discrete random variable X {\displaystyle X} can be seen as a special case of two more general measure theoretic constructions: the distribution of X {\displaystyle X} and the probability density function of X {\displaystyle X} with respect to the counting measure. We make this more precise below.
Suppose that ( A , A , P ) {\displaystyle (A,{\mathcal {A}},P)} is a probability space and that ( B , B ) {\displaystyle (B,{\mathcal {B}})} is a measurable space whose underlying σ-algebra is discrete, so in particular contains singleton sets of B {\displaystyle B} . In this setting, a random variable X : A → B {\displaystyle X\colon A\to B} is discrete provided its image is countable. The pushforward measure X ∗ ( P ) {\displaystyle X_{*}(P)} —called the distribution of X {\displaystyle X} in this context—is a probability measure on B {\displaystyle B} whose restriction to singleton sets induces the probability mass function (as mentioned in the previous section) f X : B → R {\displaystyle f_{X}\colon B\to \mathbb {R} } since f X ( b ) = P ( X − 1 ( b ) ) = P ( X = b ) {\displaystyle f_{X}(b)=P(X^{-1}(b))=P(X=b)} for each b ∈ B {\displaystyle b\in B} .
Now suppose that ( B , B , μ ) {\displaystyle (B,{\mathcal {B}},\mu )} is a measure space equipped with the counting measure μ {\displaystyle \mu } . The probability density function f {\displaystyle f} of X {\displaystyle X} with respect to the counting measure, if it exists, is the Radon–Nikodym derivative of the pushforward measure of X {\displaystyle X} (with respect to the counting measure), so f = d X ∗ P / d μ {\displaystyle f=dX_{*}P/d\mu } and f {\displaystyle f} is a function from B {\displaystyle B} to the non-negative reals. As a consequence, for any b ∈ B {\displaystyle b\in B} we have P ( X = b ) = P ( X − 1 ( b ) ) = X ∗ ( P ) ( b ) = ∫ b f d μ = f ( b ) , {\displaystyle P(X=b)=P(X^{-1}(b))=X_{*}(P)(b)=\int _{b}fd\mu =f(b),}
demonstrating that f {\displaystyle f} is in fact a probability mass function.
When there is a natural order among the potential outcomes x {\displaystyle x} , it may be convenient to assign numerical values to them (or n-tuples in case of a discrete multivariate random variable) and to consider also values not in the image of X {\displaystyle X} . That is, f X {\displaystyle f_{X}} may be defined for all real numbers and f X ( x ) = 0 {\displaystyle f_{X}(x)=0} for all x ∉ X ( S ) {\displaystyle x\notin X(S)} as shown in the figure.
The image of X {\displaystyle X} has a countable subset on which the probability mass function f X ( x ) {\displaystyle f_{X}(x)} is one. Consequently, the probability mass function is zero for all but a countable number of values of x {\displaystyle x} .
The discontinuity of probability mass functions is related to the fact that the cumulative distribution function of a discrete random variable is also discontinuous. If X {\displaystyle X} is a discrete random variable, then P ( X = x ) = 1 {\displaystyle P(X=x)=1} means that the casual event ( X = x ) {\displaystyle (X=x)} is certain (it is true in 100% of the occurrences); on the contrary, P ( X = x ) = 0 {\displaystyle P(X=x)=0} means that the casual event ( X = x ) {\displaystyle (X=x)} is always impossible. This statement isn't true for a continuous random variable X {\displaystyle X} , for which P ( X = x ) = 0 {\displaystyle P(X=x)=0} for any possible x {\displaystyle x} . Discretization is the process of converting a continuous random variable into a discrete one.
Examples
Main articles: Bernoulli distribution, Binomial distribution, and Geometric distribution
Finite
There are three major distributions associated, the Bernoulli distribution, the binomial distribution and the geometric distribution.
- Bernoulli distribution: ber(p) , is used to model an experiment with only two possible outcomes. The two outcomes are often encoded as 1 and 0. p X ( x ) = { p , if x is 1 1 − p , if x is 0 {\displaystyle p_{X}(x)={\begin{cases}p,&{\text{if }}x{\text{ is 1}}\\1-p,&{\text{if }}x{\text{ is 0}}\end{cases}}} An example of the Bernoulli distribution is tossing a coin. Suppose that S {\displaystyle S} is the sample space of all outcomes of a single toss of a fair coin, and X {\displaystyle X} is the random variable defined on S {\displaystyle S} assigning 0 to the category "tails" and 1 to the category "heads". Since the coin is fair, the probability mass function is p X ( x ) = { 1 2 , x = 0 , 1 2 , x = 1 , 0 , x ∉ { 0 , 1 } . {\displaystyle p_{X}(x)={\begin{cases}{\frac {1}{2}},&x=0,\\{\frac {1}{2}},&x=1,\\0,&x\notin \{0,1\}.\end{cases}}}
- Binomial distribution, models the number of successes when someone draws n times with replacement. Each draw or experiment is independent, with two possible outcomes. The associated probability mass function is ( n k ) p k ( 1 − p ) n − k {\textstyle {\binom {n}{k}}p^{k}(1-p)^{n-k}} . An example of the binomial distribution is the probability of getting exactly one 6 when someone rolls a fair die three times.
- Geometric distribution describes the number of trials needed to get one success. Its probability mass function is p X ( k ) = ( 1 − p ) k − 1 p {\textstyle p_{X}(k)=(1-p)^{k-1}p} .An example is tossing a coin until the first "heads" appears. p {\displaystyle p} denotes the probability of the outcome "heads", and k {\displaystyle k} denotes the number of necessary coin tosses. Other distributions that can be modeled using a probability mass function are the categorical distribution (also known as the generalized Bernoulli distribution) and the multinomial distribution.
- If the discrete distribution has two or more categories one of which may occur, whether or not these categories have a natural ordering, when there is only a single trial (draw) this is a categorical distribution.
- An example of a multivariate discrete distribution, and of its probability mass function, is provided by the multinomial distribution. Here the multiple random variables are the numbers of successes in each of the categories after a given number of trials, and each non-zero probability mass gives the probability of a certain combination of numbers of successes in the various categories.
Infinite
The following exponentially declining distribution is an example of a distribution with an infinite number of possible outcomes—all the positive integers: Pr ( X = i ) = 1 2 i for i = 1 , 2 , 3 , … {\displaystyle {\text{Pr}}(X=i)={\frac {1}{2^{i}}}\qquad {\text{for }}i=1,2,3,\dots } Despite the infinite number of possible outcomes, the total probability mass is 1/2 + 1/4 + 1/8 + ⋯ = 1, satisfying the unit total probability requirement for a probability distribution.
Multivariate case
Main article: Joint probability distribution
Two or more discrete random variables have a joint probability mass function, which gives the probability of each possible combination of realizations for the random variables.
Further reading
- Johnson, N. L.; Kotz, S.; Kemp, A. (1993). Univariate Discrete Distributions (2nd ed.). Wiley. p. 36. ISBN 0-471-54897-9.
References
7.2 - Probability Mass Functions | STAT 414 - PennState - Eberly College of Science https://online.stat.psu.edu/stat414/lesson/7/7.2 ↩
Stewart, William J. (2011). Probability, Markov Chains, Queues, and Simulation: The Mathematical Basis of Performance Modeling. Princeton University Press. p. 105. ISBN 978-1-4008-3281-1. 978-1-4008-3281-1 ↩
A modern introduction to probability and statistics : understanding why and how. Dekking, Michel, 1946-. London: Springer. 2005. ISBN 978-1-85233-896-1. OCLC 262680588.{{cite book}}: CS1 maint: others (link) 978-1-85233-896-1 ↩
A modern introduction to probability and statistics : understanding why and how. Dekking, Michel, 1946-. London: Springer. 2005. ISBN 978-1-85233-896-1. OCLC 262680588.{{cite book}}: CS1 maint: others (link) 978-1-85233-896-1 ↩
Rao, Singiresu S. (1996). Engineering optimization : theory and practice (3rd ed.). New York: Wiley. ISBN 0-471-55034-5. OCLC 62080932. 0-471-55034-5 ↩