Créer des listes en utilisant yield dans Ruby et Python

Question

J'essaie de trouver une manière élégante de créer une liste à partir d'une fonction qui génère des valeurs dans Python et Ruby.

En Python:

def foo(x): 
    for i in range(x): 
     if bar(i): yield i 
result = list(foo(100)) 

Dans Ruby:

def foo(x) 
    x.times {|i| yield i if bar(i)} 
end 
result = [] 
foo(100) {|x| result << x} 

Bien que j'aime travailler dans les deux langues, je l'ai toujours été un peu dérangé par la version Ruby ayant pour initialiser la liste et Puis remplissez-le. Les résultats yield de Python donnent une itération simple, ce qui est génial. Le yield de Ruby invoque un bloc, ce qui est aussi génial, mais quand je veux juste remplir une liste, ça me semble un peu maladroit.

Y a-t-il une manière plus élégante de Ruby?

MISE À JOUR Retravaillé l'exemple pour montrer que le nombre de valeurs générées par la fonction n'est pas nécessairement égal à x.

Answer 1

Alors, pour votre nouvel exemple, essayez ceci:

def foo(x) 
    (0..x).select { |i| bar(i) } 
end 

Fondamentalement, sauf si vous écrivez un itérateur de votre propre, vous n'avez pas besoin yield très souvent dans Ruby. Vous ferez probablement beaucoup mieux si vous arrêtez d'essayer d'écrire des idiomes Python en utilisant la syntaxe Ruby.

Answer 2

Je sais que ce n'est pas exactement ce que vous recherchez, mais plus élégante façon d'exprimer votre exemple en rubis est:

result = Array.new(100) {|x| x*x} 

Answer 3

def squares(x) 
    (0..x).map { |i| i * i } 
end 

Tout ce qui concerne une gamme de valeurs est meilleure façon de traiter avec, bien , une plage, plutôt que times et la génération de tableau.

Answer 4

Pour la version Python j'utiliser une expression de générateur comme:

(i for i in range(x) if bar(i)) 

Ou pour ce cas spécifique des valeurs de filtrage, encore plus simplement

itertools.ifilter(bar,range(x)) 

Answer 5

Pour la liste Python version compréhension affichée par stbuton utiliser xrange au lieu de la gamme si vous voulez un générateur. range va créer toute la liste en mémoire.

Answer 6

yield signifie différentes choses ruby ​​et python. Dans ruby, vous devez spécifier un bloc de rappel si je me souviens bien, alors que les générateurs en python peuvent être transmis et céder à celui qui les détient.

Answer 7

L'équivalent exact de votre code Python (en utilisant des générateurs Ruby) serait:

def foo(x) 
    Enumerator.new do |yielder| 
     (0..x).each { |v| yielder.yield(v) if bar(v) } 
    end 
end 

result = Array(foo(100)) 

Dans ce qui précède, la liste est générée paresseusement (comme dans l'exemple Python); voir:

def bar(v); v % 2 == 0; end 

f = foo(100) 
f.next #=> 0 
f.next #=> 2