numpy.random.choice()函数详解

numpy.random.choice(a, size=None, replace=True, p=None)函数从指定随机选择元素。

此函数返回给定大小的新数组，其元素来自在没有替换的情况下从输入数组中随机选择的元素。

参数：

1. a: 一维数组或整数，表示可选项。如果是一个整数，则该整数是范围从零到n-1的一维数组。
2. size: 可选参数，输出的形状。指定要返回的数组的形状。如果是一个整数或元组，则其表示输出数组的大小。如果None，则返回单个值。
3. replace: 可选参数，表示是否允许有重复。如果是True，则生成的样本可以重复。如果是False，则生成的样本不会重复。
4. p: 可选参数，表示概率分布。一维数组中的每个元素被视为从该数组中选择相应索引的概率。数组必须具有与a相同的长度。

返回值：

大小为“size”的新数组，其元素来自在没有替换的情况下随机选择的a中的元素。

如果“a”是整数，则返回范围从0到n-1的一维数组。如果指定了“p”，则a中每个元素的概率为p。

示例1：从一维数组中随机选择一个元素

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(np.random.choice(arr))

输出：

2

示例2：生成一个不重复的大小为3的数组

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
print(np.random.choice(arr, size=3, replace=False))

输出：

[9 8 7]

注意：由于“replace”参数设置为False，因此生成的样本不会重复。如果设置为True，则生成的样本可以重复。

以上就是numpy.random.choice()函数的详细说明，希望可以帮助你更好地理解和使用这个函数。