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)