【python】np.random.choice()

最新推荐文章于 2025-05-28 23:37:43 发布

原创

最新推荐文章于 2025-05-28 23:37:43 发布 · 1.5k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文介绍了numpy.random.choice()函数的使用，该函数用于从一维数组中随机抽取元素，可选择是否允许重复，并可以自定义每个元素的抽取概率。详细讲解了replace参数（决定是否允许抽取相同元素）和p参数（设定抽取概率）。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

np.random.choice()：

numpy.random.choice(a, size=None, replace=True, p=None)

从a(只要是ndarray都可以，但必须是一维的)中随机抽取数字，并组成指定大小(size)的数组
replace:True表示可以取相同数字，False表示不可以取相同数字
数组p：与数组a相对应，表示取数组a中每个元素的概率，默认为选取每个元素的概率相同。

产生随机数：

>>>np.random.choice(5)#从[0, 5)中随机输出一个随机数
#相当于np.random.randint(0, 5)
	2

>>>np.random.choice(5, 3)#在[0, 5)内输出五个数字并组成一维数组（ndarray）
#相当于np.random.randint(0, 5, 3)
	array([1, 4, 1])

从数组、列表或元组中随机抽取

注意：不管是什么，它必须是一维的！

L = [1, 2, 3, 4, 5]#list列表
T = (2, 4, 6, 2)#tuple元组
A = np.array([4, 2, 1])#numpy,array数组,必须是一维的
A0 = np.arange(10).reshape(2, 5)#二维数组会报错

>>>np.random.choice(L, 5)
	array([3, 5, 2, 1, 5])
	
>>>np.random.choice(T, 5)
	array([2, 2, 2, 4, 2])
 
>>>np.random.choice(A