【链接】Legacy Random Generation — NumPy v1.22 Manual
一些函数的使用教程1.1 random.randint(low, high=None, size=None, dtype=int)
【功能】返回指定闭区间内的整数
【参数】
high,low:指定整数的取值上下限(闭区间),
size:指定输出数据的shape,比如:size=(m,n,k),则函数返回(m,n,k)个整数
【参考文档】
numpy.random.randint — NumPy v1.22 Manual
1.2 random.choice(a, size=None, replace=True, p=None)
【功能】根据指定一维数组(a)或一个整数,生成一个随机样本
【参数】
a: 传入1-D数组或一个整数
若a的位置传入的是1-D数组(array),则使用该数组的元素随机生成新的样本(shape由size传入参数指定);若传入整数,则指定输出的取值范围为np.arange(a)
size:输出数据的形状
p:当a传入的是1-D array时,p处可以传入相同size的数组,p中各元素与a的元素对应,表示函数生成a中该元素的概率。
【参考文档】
numpy.random.choice — NumPy v1.22 Manual
【例子】
1 a传入一个数组
a为一个字符串数组,p指定了各字符串生成的概率,size为输出的shape
可以看到输出了一个大小为5的1-D array,且有三个元素都是'pooh'(因为其概率最高,为0.5)
2 a传入一个整数
a=5为整数,则输出元素的取值范围为[0,1,2,3,4],输出3的概率为0.3
size=3,则output为3个整数组成的array