np.random.randint()函数的功能和用法
时间: 2023-10-13 07:56:15 浏览: 162
np.random.randint()函数是NumPy库中的一个函数,用于生成随机整数。
使用示例:
```python
import numpy as np
# 生成一个随机整数
random_int = np.random.randint(low, high=None, size=None, dtype=int)
print(random_int)
```
该函数的参数包括:
- low:生成的随机整数的最小值(包含),可以是整数或数组。如果只提供一个参数,则被视为最大值,并默认最小值为0。
- high:生成的随机整数的最大值(不包含),可以是整数或数组。如果未提供此参数,则范围为[0, low)。
- size:生成的随机整数的形状,可以是一个整数或一个表示形状的元组。默认为None,表示生成单个随机整数。
- dtype:生成的随机整数的数据类型,可以是int、int8、int16等。默认为int。
下面是一些使用np.random.randint()函数的示例:
```python
import numpy as np
# 生成一个介于0到9之间的随机整数
random_int = np.random.randint(10)
print(random_int) # 输出一个0到9之间的随机整数
# 生成一个介于5到10之间的随机整数
random_int = np.random.randint(5, 10)
print(random_int) # 输出一个5到9之间的随机整数
# 生成一个形状为(2, 3)的随机整数数组
random_array = np.random.randint(1, 10, size=(2, 3))
print(random_array) # 输出一个2行3列的随机整数数组
```
在上面的示例中,我们使用np.random.randint()函数生成了随机整数。第一个示例中,我们没有指定最小值和最大值,因此生成的随机整数范围是0到9。第二个示例中,我们指定了最小值为5和最大值为10,因此生成的随机整数范围是5到9。第三个示例中,我们指定了最小值为1,最大值为10,并指定了形状为(2, 3),因此生成了一个2行3列的随机整数数组。
阅读全文