【数据分析之Numpy】Numpy随机抽样函数random.choice()的使用方法

一、简介

        numpy.random.choice() 是 NumPy 库中的一个函数，用于从给定的数组或整数范围中随机选择一个元素或数组。这个函数可以用于生成随机数、随机选择数组中的元素等。

二、基本语法

        函数的基本语法如下：

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

        参数说明

a：一个整数或数组，表示要从中选择元素的范围。如果是整数，表示要从 0 到 a-1 中选择一个元素。如果是数组，则返回的元素将从该数组中随机选择。

size：表示要生成的随机数的数量。如果未指定，则返回一个随机数。

replace：表示是否允许选择已选择的元素。如果为 True，则允许重复选择；如果为 False，则不允许重复选择。默认值为 True。

p：表示每个元素被选择的概率。如果指定了 p 参数，则返回的元素将按照指定的概率分布进行选择。

三、用法

1、随机选择一个或多个整数

        （1）随机选择一个整数

import numpy as np

random_int = np.random.choice(10)

print(random_int) # 输出: 一个从 0 到 9 的随机整数

        （2）随机选择多个整数

random_some_int = np.random.choice(10, 3)

print(random_some_int) # 输出: 3个从 0 到 9 的随机整数

2、随机选择一个或多个元素

        （1）随机选择一个元素

random_element = np.random.choice([1, 2, 3])

print(random_element) # 输出: 1 或 2 或 3 中的任意一个

        （2）随机选择4个元素

random_some_element = np.random.choice([1, 2, 3, 4, 5, 6, 7, 8, 9], 4)

print(random_some_element) # 输出: 4个任意元素

3、随机选择一个或多个元素，不允许重复选择

random_some_element = np.random.choice([1, 2, 3, 4, 5, 6, 7, 8, 9], 4, replace=False)

print(random_some_element) # 输出: 4个任意元素

4、随机选择一个或多个元素，指定每个元素选中的概率

random_some_element = np.random.choice([1, 2, 3, 4, 5], 3, replace=False, p=[0.1, 0.2, 0.2, 0.4, 0.1])

print(random_some_element) # 输出: 4个任意元素

注意：这里p参数内的概率加起来必须等于1，不然会报错。