Numpy生成随机整数

Question

有没有办法生成整数/整数？ 我试图找到一种方法来生成一个随机数，它要么是 0 要么是 1 要么是 2 .... 中间没有任何东西。 如果我使用：

import numpy as np

for i in range(5):
    action = np.random.uniform(0,2,1)#take random action between 0 1 or 2
    print(action)

返回：

[0.18429578]
[1.19763353]
[1.93240698]
[1.44706511]
[0.31182739]

numpy.random.randint似乎可以工作，但它会返回一个数组。 某种功能是使这项工作的最佳方法吗？ 谢谢

Answer 1

我建议您使用random模块并使用random.randint生成一个随机整数

import random 

number = random.randint(0,2)

Answer 2

用这个，

for i in range(5):
    action = np.random.randint(0,3)
    print(action)