DP Mixture Alg#2 python