façon pythonique de choisir une valeur aléatoire qui satisfait un certain prédicat

Question

Supposons que j'aie une liste d'éléments et que je veuille sélectionner aléatoirement un élément de la liste qui satisfait un prédicat. Quelle est la manière pythonique de le faire?

Je fais actuellement une compréhension suivie d'une random.choice() mais qui est inutilement inefficace:

intlist = [1,2,3,4,5,6,7,8,9] 
evenlist = [ i for i in intlist if i % 2 == 0 ] 
randomeven = random.choice(evenlist) 

Merci!

Answer 1

La façon dont vous l'avez écrit ci-dessus est en fait bon python idiomatiques. Si nous analysons l'algorithme, nous verrons qu'il fait essentiellement ceci:

1. Faire une liste d'éléments qui satisfont le prédicat. (Croît linéairement avec n)
2. Choisir un élément aléatoire dans cette liste. (Temps constant)

La seule autre façon de procéder serait de choisir un élément au hasard, de décider s'il satisfait le prédicat, et de choisir à nouveau si ce n'est pas le cas. Cet algorithme est un peu plus complexe. Dans le cas où 90% de la liste satisfait le prédicat, cela fonctionnera beaucoup plus vite que votre solution. Dans le cas où seulement 10% de la liste satisfait le prédicat, il fonctionnera beaucoup plus lentement, car il y a de fortes chances qu'il choisisse aléatoirement un élément donné et vérifie si le prédicat est satisfait sur cet élément plus d'une fois. Maintenant, vous pouvez envisager de mémoriser votre prédicat, mais vous allez toujours sélectionner beaucoup de données aléatoires. Cela se résume à ceci: à moins que votre solution ne soit particulièrement inadaptée à vos données, restez-y, parce que c'est génial. Personnellement, je réécris comme ceci:

intlist = range(1,10) 
randomeven = random.choice([i for i in intlist if i % 2 == 0]) 

C'est un peu plus concis, mais il va courir exactement le même que votre code existant.

Answer 2

Je ne pouvais pas trouver une sorte de fonction de random.selectspecific(list, predicate) dans la documentation, donc je voudrais essayer quelque chose comme ce qui suit:

import random 
def selectspecific(l, predicate): 
    result = random.choice(l) 
    while (not predicate(result)): 
     result = random.choice(l) 
    return result 

Answer 3

import random 

intlist = [1,2,3,4,5,6,7,8,9] 
randomeven = random.choice(filter(lambda x: x % 2 == 0, intlist))                                       

Answer 4

Comment pythonic est-ce?

from itertools import ifilterfalse 
from random import choice 
print choice([ i for i in ifilterfalse(lambda x: x%2, range(10)) ]) 

Answer 5

Si vous souhaitez encapsuler davantage, vous pouvez créer une méthode pour gérer la sélection, qui accepterait une méthode de prédicat. Vous pouvez ensuite utiliser cette méthode de prédicat avec filter().

import random 
def selectSpecific(intlist, predicate): 
    filteredList = filter(predicate, intlist) 
    result = random.choice(filteredList) 
    return result 

Vous pouvez passer un prédicat à selectSpecific() comme lambda ou toute autre méthode. Exemple:

intlist = range(1,10) 
selectSpecific(intlist, lambda x: x % 2 == 0) 

def makeEven(n): 
    if n % 2 == 0: 
    return n 

selectSpecific(intlist, makeEven)