discrete probability distribution python