numpy抽样函数 np.random.choice用法详解

顾名思义,抽样函数,定义如下:

def choice(a, size=None, replace=True, p=None):

参数说明:

a :待抽样的样本(一维数组或整数)
size: 输出大小,默认返回单个元素
replace : 抽样后的元素是否可重复,默认是
p: 每个样本点被抽样的概率,默认均匀抽样

举例如下:

从[1,2,3,4,5]中随机抽三个元素,可重复,概率分别为[0.1,0.1,0.2,0.1,0.5]

>>> a=[1,2,3,4,5]
>>> p=[0.1,0.1,0.2,0.1,0.5]
>>> np.random.choice(a,3,True,p)
array([5, 2, 5])

元素不可重复(即第三个参数replace设为false):

>>> np.random.choice(a,3,False,p)
array([2, 3, 5])

若输入a为整型,则表示从0到a-1中的整数样本进行抽样,如:

>>> np.random.choice(5,3,True)
array([4, 1, 3])
>>> np.random.choice(5,3,True)
array([0, 4, 2])

第二个参数size不设置,则只返回单个元素,如:

>>> np.random.choice(5)
2
>>> np.random.choice(5)
1

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2022年5月23日
下一篇 2022年5月23日

相关推荐