new_z = numpy.random.multinomial(1, p_z / p_z.sum()).argmax() multinomialは（第１引数）回だけ、（第二引数）確率の試行を繰り返し行って、その時の回数を求めるかな？で、その引数に対して最大値を求めれば「一番選択されうるもの」になると。