NumPy.random.choice()函数
NumPy.random.choice()函数是NumPy库中的一个函数，它可以从一个给定的数组中随机抽取元素，并以指定的概率返回它们。
使用示例
import numpy as np # 定义一个数组 array = np.array([1, 2, 3, 4, 5]) # 从数组中抽取一个元素 choice = np.random.choice(array) # 结果 print(choice)
详细解析
NumPy.random.choice()函数的语法如下：
numpy.random.choice(a, size=None, replace=True, p=None)
- a： 一维数组，用于从中抽取元素
- size：抽取元素的数量，默认为1
- replace：抽取元素是否可以重复，默认为True，表示可以重复抽取
- p：抽取元素的概率，如果设置了这个参数，则每个元素抽取的概率就是p中指定的值。
NumPy.random.choice()函数的返回值是一个一维数组，其中包含了从a中抽取的元素。
实例
下面是一个使用NumPy.random.choice()函数从数组中抽取3个元素的实例：
import numpy as np # 定义一个数组 array = np.array([1, 2, 3, 4, 5]) # 从数组中抽取3个元素 choice = np.random.choice(array, size=3) # 结果 print(choice)
运行结果如下：
[3 4 5]
下面是一个使用NumPy.random.choice()函数从数组中抽取3个元素，且每个元素抽取的概率不相同的实例：
import numpy as np # 定义一个数组 array = np.array([1, 2, 3, 4, 5]) # 每个元素抽取的概率 p = np.array([0.1, 0.2, 0.3, 0.2, 0.2]) # 从数组中抽取3个元素 choice = np.random.choice(array, size=3, p=p) # 结果 print(choice)
运行结果如下：
[2 5 3]