np.random.choice的含义和用法
时间: 2024-04-07 17:25:41 浏览: 17
np.random.choice是NumPy库中的一个函数,用于从给定的一维数组中随机选择元素。它的语法如下:
np.random.choice(a, size=None, replace=True, p=None)
参数说明:
- a:一维数组,表示要进行选择的元素集合。
- size:整数或者整数元组,表示要选择的元素个数。如果为None,则返回单个随机元素;如果为整数,则返回一个包含指定个数元素的一维数组;如果为整数元组,则返回一个包含指定形状的数组。
- replace:布尔值,表示是否允许重复选择。如果为True,则允许重复选择;如果为False,则不允许重复选择。
- p:一维数组,表示每个元素被选择的概率。如果为None,则默认每个元素被选择的概率相等。
示例用法:
```python
import numpy as np
# 从一维数组中随机选择一个元素
x = np.array([1, 2, 3, 4, 5])
random_element = np.random.choice(x)
print(random_element)
# 从一维数组中随机选择3个元素,允许重复选择
random_elements = np.random.choice(x, size=3, replace=True)
print(random_elements)
# 从一维数组中随机选择3个元素,不允许重复选择
random_elements = np.random.choice(x, size=3, replace=False)
print(random_elements)
# 从一维数组中按指定概率随机选择一个元素
random_element = np.random.choice(x, p=[0.1, 0.2, 0.3, 0.2, 0.2])
print(random_element)
```