In reinforcement learning, I got many discrete distributions corresponding to different states, like the following:
import numpy as np
distributions = np.array([[0.1,0.2,0.7],[0.3,0.3,0.4],[0.2,0.2,0.6]])
# array([[0.1, 0.2, 0.7], # \pi(s0)
# [0.3, 0.3, 0.4], # \pi(s1)
# [0.2, 0.2, 0.6]]) # \pi(s2)