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

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)数组中取样。
    其他的用到了再补充。