Probability mass function


A probability mass function (PMF) is a function associated with a discrete Random variable that gives the Probability that the variable, when measured, is exactly equal to some value. The PMF describes the Probability distribution that a discrete variable follows.

Given a discrete random variable XX, its probability mass function pXp_{X} is defined as pX:ΩX[0,1]p_{X}:\Omega_{X}\mapsto[0,1], where ΩX\Omega_{X} is XX's sample space. It is equal to

pX(x)=P(X=x)p_{X}(x)=P(X=x)

where PP is a measure of probability. By definition of probability, the total probability must be 1:

xΩpX(x)=1\sum_{x \in \Omega}p_{X}(x)=1

Transformations

Given a discrete random variable XX of Probability mass function fXf_{X} and an invertible transformation g(x)g(x), we can define the transformed random variable as Y=g(X)Y=g(X). The PMF of YY is

fY(y)=fX(g1(y))f_{Y}(y)=f_{X}(g^{-1}(y))

The same definition applies to a discrete random vector X\mathbf{X} that is invertibly transformed into another random vector Y=g(X)\mathbf{Y}=g(\mathbf{X}):

fY(y)=fX(g1(y))f_{\mathbf{Y}}(\mathbf{y})=f_{\mathbf{X}}(g^{-1}(\mathbf{y}))