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)