潇洒月饼

文章
8
资源
1
加入时间
3年1月18天

Numpy中的random.choice()函数 随机选择

import numpy as nptemp = np.array([[0.8, 0.2]])print(temp.shape)action = np.random.choice(2, p=temp.ravel())# ravel从一维度降到二维度。print(action)# 0,1;0.8,0.2(以这两个概率选择0和1)语法介绍:numpy.random.choice(a, size=None, replace=True, p=None)a:如果a是一个一维数组,则就是从这个数组中进行