ruby-on-rails - 如何在 ruby​​ 中使用 next 和 inject

Question

我正在编写如下代码:

[1,2,3,4,5].inject([]) do |res, a|
  res << a*a and next if a == 2
  res << a
end

出现以下错误:

NoMethodError: undefined method `<<' for nil:NilClass

接下来它生成 res变量为 nil , 如何解决这个问题？

我尝试了各种方法，但无法使用 ruby​​，我知道我提供的这个片段可以在没有 next( a == 4 ? res << a*a : res << a ) 的情况下完成，但在我的实际用例中，我有一些复杂的逻辑并且可以就这么简单。

Answer 1

替换

res << a*a and next if a == 2

与

next res << a*a if a == 2

现在，它会起作用。

例子:-

#!/usr/bin/env ruby

ar = [1,2,3,4,5].inject([]) do |res, a|
  next res << a*a if a == 2
  res << a
end

p ar
# >> [1, 4, 3, 4, 5]

Read the documentation of next

next can take a value, which will be the value returned for the current iteration of the block....