随机获取用np.random.choice 指定获取用什么

np.random.choice函数可以指定从哪个序列中随机获取元素，也可以指定获取的数量、是否可重复获取、元素的权重等参数。以下是一些示例：

1. 从列表中随机选择一个元素：

import numpy as np

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

输出：

3

2. 从数组中随机选择多个元素：

import numpy as np

my_array = np.array([1, 2, 3, 4, 5])
random_elements = np.random.choice(my_array, size=3, replace=False)
print(random_elements)

输出：

[2 4 1]

3. 从字典中随机选择一个键：

import numpy as np

my_dict = {'a': 1, 'b': 2, 'c': 3, 'd': 4, 'e': 5}
random_key = np.random.choice(list(my_dict.keys()))
print(random_key)

输出：

d

4. 按照元素的权重进行随机选择：

import numpy as np

my_array = np.array([1, 2, 3, 4, 5])
weights = np.array([0.1, 0.2, 0.3, 0.2, 0.2])
random_element = np.random.choice(my_array, p=weights)
print(random_element)

输出：

3