python中 np.random.choice()

np.random.choice() 是 NumPy 库中的一个函数，用于从给定的一维数组中随机选择元素。它的语法如下：

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

参数说明：
- a：一维数组或整数，表示要进行选择的元素。如果是整数，则从0到该整数（不包括该整数）之间进行选择。
- size：整数或元组，表示返回的样本的大小。默认为 None，表示返回单个随机元素。
- replace：布尔值，表示是否允许重复选择元素。默认为 True，即允许重复选择。
- p：一维数组，表示每个元素被选择的概率。默认为 None，表示每个元素有相等的被选择概率。

示例用法：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
choice = np.random.choice(arr)
print(choice)  # 随机选择 arr 中的一个元素

choices = np.random.choice(arr, size=3)
print(choices)  # 随机选择 arr 中的三个元素

choices = np.random.choice(arr, size=(2, 2))
print(choices)  # 随机选择 arr 中的两个元素，并构成一个二维数组

weights = np.array([0.1, 0.2, 0.3, 0.2, 0.2])
choice = np.random.choice(arr, p=weights)
print(choice)  # 按照权重随机选择 arr 中的一个元素

希望能对你有所帮助！如果还有其他问题，请继续提问。