¿Existe una forma sencilla en Python de generar un número aleatorio en un rango que excluya algún subconjunto de números en ese rango?
Por ejemplo, sé que puedes generar un número aleatorio entre 0 y 9 con:
from random import randint randint(0,9) ¿Qué sucede si tengo una lista, por ejemplo, exclude=[2,5,7] , que no quiero que se devuelva?
Prueba esto:
from random import choice print choice([i for i in range(0,9) if i not in [2,5,7]])Prueba con algo como esto:
from random import randint def random_exclude(*exclude): exclude = set(exclude) randInt = randint(0,9) return my_custom_random() if randInt in exclude else randInt print(random_exclude(2, 5, 7))Si tiene listas más grandes, recomendaría usar operaciones establecidas porque se notan más rápido que la respuesta recomendada.
random.choice(list(set([x for x in range(0, 9)]) - set(to_exclude)))Tomé algunas pruebas con la respuesta aceptada y mi código anterior.
Para cada prueba hice 50 iteraciones y medí el tiempo promedio. Para las pruebas utilicé un rango de 999999.
para_excluir elementos de tamaño 10:
Respuesta aceptada = 0.1782s
Esta respuesta = 0.0953s
para_excluir elementos de tamaño 100:
Respuesta aceptada = 01.2353s
Esta respuesta = 00.1117s
para_excluir elementos de tamaño 1000:
Respuesta aceptada = 10.4576s
Esta respuesta = 00.1009s