Search code examples
pythonpython-3.xrandomsampling

How to generate a random 4 digit number not starting with 0 and having unique digits?


This works almost fine but the number starts with 0 sometimes:

import random
numbers = random.sample(range(10), 4)
print(''.join(map(str, numbers)))

I've found a lot of examples but none of them guarantee that the sequence won't start with 0.


Solution

  • We generate the first digit in the 1 - 9 range, then take the next 3 from the remaining digits:

    import random
    
    # We create a set of digits: {0, 1, .... 9}
    digits = set(range(10))
    # We generate a random integer, 1 <= first <= 9
    first = random.randint(1, 9)
    # We remove it from our set, then take a sample of
    # 3 distinct elements from the remaining values
    last_3 = random.sample(digits - {first}, 3)
    print(str(first) + ''.join(map(str, last_3)))
    

    The generated numbers are equiprobable, and we get a valid number in one step.