np.random.choice（）怎么使用

np.random.choice() 是 NumPy 库中的一个函数，用于从给定的数组或序列中随机选择元素。

使用方式如下：

import numpy as np

# 从整数数组中随机选择一个元素
array = np.array([1, 2, 3, 4, 5])
random_element = np.random.choice(array)
print(random_element)

# 从序列中随机选择一个元素
sequence = ['a', 'b', 'c', 'd', 'e']
random_element = np.random.choice(sequence)
print(random_element)

# 随机选择多个元素
random_elements = np.random.choice(array, size=3)
print(random_elements)

# 设置概率分布进行选择，可以通过传递一个概率数组作为参数
probability = [0.1, 0.3, 0.2, 0.2, 0.2]
random_element = np.random.choice(array, p=probability)
print(random_element)

在上面的示例中，`np.random.choice()` 函数会从给定的数组或序列中随机选择一个元素，并返回所选元素。你可以设置 `size` 参数来选择多个元素，并且可以通过设置 `p` 参数来指定概率分布进行选择。