我试图想出一个优雅的方法,从一个函数创建一个列表,在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