Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How do I generate a list of random integers without consecutive numbers?

After I sort a randomly generated list, I have consecutive numbers, e.g.

[7, 9, 13, 47, 64, 76, 83, 94, 95, 114, 115, 116, 120, 121, 123, 124, 127, 136, 152, 154, 167, 184, 189, 205, 212, 222, 226, 229, 231, 238]

Here consecutive numbers are (94, 95), (120, 121) and (123, 124). How do I remove them?

My code is:

while len(set(l)) != 30:
    a = random.randint(1, 240)
    l.append(a)

l = list(set(l))
l = sorted(l)

f.write(str(l))

I do not want to use randrange or choice from random module.

like image 223
Abdemanaaf Ghadiali Avatar asked Oct 16 '22 08:10

Abdemanaaf Ghadiali


1 Answers

Create a random number a and check if a and a±1 are not in the set:

import random 
l = set()
while len(l) != 30:
    a = random.randint(1, 240)
    if not {a-1,a,a+1} & l: # set intersection: empty == False == no common numbers
        l.add(a)

l = sorted(l) # sorted creates a sorted list from any iterable

print(l) 

Output:

[5, 12, 40, 47, 55, 59, 62, 73, 76, 82, 84, 89, 93, 95, 109, 
 125, 127, 141, 165, 168, 184, 187, 196, 202, 204, 210, 215, 
 218, 229, 231]

Directly using a set makes the check if the number (±1) is already part of your random numbers very fast.

Doku:

  • set.intersection (or &)

And as function:

import random

def get_random_numbers_no_neighboring_elems(min_num, max_num, amount):
    """Generates amount random numbers in [min_num,..,max_num] that do not
    include neighboring numbers."""

    # this is far from exact - it is best to have about 5+ times the amount 
    # of numbers to choose from - if the margin is too small you might take
    # very long to get all your "fitting numbers" as only about 1/4 of the range
    # is a viable candidate (worst case): 
    #   [1 2 3 4 5 6 7 8 9 10]: draw 2 then 5 then 8 and no more are possible 
    if (max_num-min_num) // 5 < amount:
        raise ValueError(f"Range too small - increase given range.")

    l = set()
    while len(l) != amount:  
        a = random.randint(min_num, max_num)
        if not {a-1,a,a+1} & l: # set intersection: empty == False == no commons
            l.add(a)
    return sorted(l)

print(get_random_numbers_no_neighboring_elems(1,240,80))
like image 63
Patrick Artner Avatar answered Oct 20 '22 23:10

Patrick Artner