python - python中的多项式样本生成

Question

我能够使用以下代码行从所需大小的概率数组生成二项式样本(输出应与输入概率数组形状的大小相同)

prob_list = [[0.3,0.3,0.4],[0.4,0.3,0.3]]
prob_array = np.asarray(prob_list)
y_sample = np.random.binomial(size=prob_array.shape, n=1, p=prob_array)
print(y_sample)

输出是

[[0 0 0]
[1 1 1]]

输入概率(2*3)的形状和输出样本(2*3)的形状相同。

是否可以对多项式做同样的事情？可以为单行生成多项式样本。

y_sample = np.random.multinomial(size=1, n=1, pvals=prob_array[0])
print(y_sample)
[[1 0 0]]

如何概括这一点以获得与二项式相同的输出(输出样本形状 = 输入概率形状？

Answer 1

恐怕您将不得不手动遍历 prob_array。 pvals 目前仅限于单个一维概率数组。