numpy.random.randint

numpy.random.randint(low, high=None, size=None)

Return random integers x such that low <= x < high.

If high is None, then 0 <= x < low.