从numpy.random.multinomial获取随机的指数数组

Question

考虑下面的文档示例，提取一个骰子的20张图

>>> np.random.multinomial(20, [1/6.]*6, size=1)
array([[4, 1, 7, 5, 2, 1]])

我想要的输出是一个20维数组，其中包含采样结果的指示：

[0,0,0,0,1,2,2,2,2,2,2,2,3,3,3,3,3,4,4,5]

但不应该对它们进行排序。也就是说，我希望多项式的输出“看起来”像np.random.uniform或np.random.normal的输出，也就是随机抽取的数组。

Answer 1

你可以像这样用random.choice得到结果。其以相等的概率取值0到5的20个样本。您还可以将参数p作为数组传递给choice函数，p是每个元素的概率

np.random.choice(6, 20)