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

    科技2023-09-15  101

    import numpy as np temp = 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是一个一维数组,则就是从这个数组中进行随机取样, 若a是一个数,则从numpy.arange(a)数组中取样。 其他的用到了再补充。
    Processed: 0.023, SQL: 8