arrays - Ruby，Array.select 与元素的数量

Question

我有这个数组

[1, 2, 3, 4, 5, 6]

我想获取前 2 个大于 3 的元素。

我能做到:

elements = []

[1, 2, 3, 4, 5, 6].each do |element|
  elements << element if element > 3
  break if elements.size == 2
end

puts elements

有没有更优雅的方法来做到这一点？

Ruby 核心中是否有类似Array.select(num_elements, &block) 的东西？

Answer 1

你快到了。只需使用带参数的 break 即可:

[1, 2, 3, 4, 5, 6].each_with_object([]) do |element, acc|
  acc << element if element > 3
  break acc if acc.size >= 2
end

实现它的另一种方法是使用 Enumerator::Lazy使用 array.lazy.select，或使用 Enumerable#take 的显式 Enumerator 实例(这里绝对是矫枉过正，发布主要用于教育目的。 )

enum =
  Enumerator.new do |y|
    i = [1, 2, 3, 4, 5, 6].each
    loop { i.next.tap { |e| y << e if e > 3 } }
  end
enum.take(2)
#⇒ [4, 5]

旁注:一旦找到两个元素，上述两个示例都会停止遍历输入。