np.random.choice(选择array中概率最大的元素)

np.random.seed(0)
p = np.array([0.1, 0.0, 0.7, 0.2])
index = np.random.choice([0, 1, 2, 3], p = p.ravel())

2

posted @ 2020-06-26 01:38  douzujun  阅读(428)  评论(0编辑  收藏  举报