跳过Enumerable#collect中的迭代

Question

(1..4).collect do |x|
  next if x == 3
  x + 1
end # => [2, 3, nil, 5]
    # desired => [2, 3, 5]

如果满足条件,next则collect放入nil数组,而我正在尝试做的是,如果满足条件,则不在返回的数组中放置任何元素.这可能不调用delete_if { |x| x == nil }返回的数组吗？

(使用Ruby 1.8.7;我的代码摘录被大量抽象)

Answer 1

有一种方法Enumerable#reject可以达到目的:

(1..4).reject{|x| x == 3}.collect{|x| x + 1}

直接使用一个方法的输出作为另一个方法的输入的做法称为方法链接,在Ruby中很常见.

BTW,map(或collect)用于将可枚举的输入直接映射到输出的输入.如果您需要输出不同数量的元素,那么您可能需要另一种方法Enumerable.

编辑:如果您对某些元素被迭代两次这一事实感到困扰,您可以使用基于inject(或其类似方法命名each_with_object)的不太优雅的解决方案:

(1..4).each_with_object([]){|x,a| a << x + 1 unless x == 3}

Answer 2

我只是调用.compact结果数组,它删除数组中的任何nil实例.如果您想要修改现有阵列(没有理由不这样做),请使用.compact!:

(1..4).collect do |x|
  next if x == 3
  x
end.compact!

Answer 3

红宝石 2.7+

现在有！

Ruby 2.7 正是filter_map为此目的而引入的。这是惯用的和高性能的，我希望它很快成为常态。

例如：

numbers = [1, 2, 5, 8, 10, 13]
numbers.filter_map { |i| i * 2 if i.even? }
# => [4, 16, 20]

这是有关该主题的好读物。

希望这对某人有用！