使用Ruby和Python中的yield创建列表

我试图想出一个优雅的方法,从一个函数创建一个列表,在Python和Ruby中产生值.

在Python中:

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

在Ruby中:

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

虽然我喜欢用两种语言工作,但我总是对Ruby版本有点困扰,不得不初始化列表然后填写它.Python的yield结果是简单的迭代,这很棒.Ruby会yield调用一个块,这也很棒,但是当我只想填写一个列表时,感觉有点笨重.

有更优雅的Ruby方式吗？

更新重写示例以显示函数产生的值的数量不一定等于x.

所以,对于你的新例子,试试这个:

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

基本上,除非您正在编写自己的迭代器,否则yield在Ruby中不需要经常使用它.如果你不再尝试使用Ruby语法编写Python习语,你可能会做得更好.

对于Python版本,我将使用生成器表达式,如:

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

或者对于这种过滤值的特定情况,甚至更简单

itertools.ifilter(bar,range(x))

您的Python代码(使用Ruby Generators)的完全等价物将是:

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

result = Array(foo(100))

在上面,列表是懒惰生成的(就像在Python示例中一样); 看到:

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

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