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

发布时间 2023-04-11 18:02:38作者: 无左无右
import numpy as np
import random
list_a = ["a", "b", "c", "d", "e", "f", "g"]

get_ = np.random.choice(list_a, 4, replace=False)
print(get_)

输出:

['a' 'f' 'c' 'e']

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

>>> aa_milne_arr = ['pooh', 'rabbit', 'piglet', 'Christopher']
>>> np.random.choice(aa_milne_arr, 5, p=[0.5, 0.1, 0.1, 0.3])
	array(['pooh', 'pooh', 'pooh', 'Christopher', 'piglet'], dtype='|S11')
#可以看到,‘pooh’被选取的概率明显比其他几个高很多