Ruby：根据块查找条件删除并返回数组值

Question

是否有一种内置方法可以根据返回 true 的块条件从数组中删除值，并返回被删除的值？

这是我正在尝试做的简化版本，但似乎必须有更好的方法：

array = [1,2,3,4,5,6,7,8,9,10]

index = array.index {|v| v == 5} # returns index if block is true

value = array.delete_at(index) # deletes and returns element at index

value 则为 5

Answer 1

您无法就地更新数组并返回一组已删除的不同值。您可以使用delete_if 执行以下操作来删除值并捕获块中的逻辑删除的值：

reject = []
 => [] 
content = [1,2,3,4,5,6,7,8,9]
 => [1, 2, 3, 4, 5, 6, 7, 8, 9] 
content.delete_if {|v| reject << v if v > 5}
 => [1, 2, 3, 4, 5] 
reject
 => [6, 7, 8, 9] 
content
 => [1, 2, 3, 4, 5] 

Answer 2

您真的需要从原始数组中删除项目吗？ 你真的只是想根据一些把它分成两部分 健康）状况？如果是后者，那么：

accepted = [ ]
rejected = [ ]
original.each { |e| (want_this_one(e) ? accepted : rejected).push(e) }

或

parts = original.inject({ :accepted => [ ], :rejected => [ ] }) do |accumulator, e|
  if(want_this_one(e))
    accumulator[:accepted].push(e)
  else
    accumulator[:rejected].push(e)
  end
  accumulator
end

然后是一个简单的方法包装器，可以很容易地提供一个块：

def categorize(array)
  categories = array.inject({ :accepted => [ ], :rejected => [ ] }) do |accumulator, e|
    if(yield e)
      accumulator[:accepted].push(e)
    else
      accumulator[:rejected].push(e)
    end
    accumulator
  end
  return categories[:accepted], categories[:rejected]
end

kept, deleted = categorize([1, 2, 3, 4, 5]) { |n| n % 2 == 0 }
# kept    = [2, 4]
# deleted = [1, 3, 5]

或者您可以使用Enumerable#partition 将数组拆分为两部分。

如果您确实需要就地修改数组，那么这个版本的 Wes 应该可以解决问题：

def slice_out(array)
  dead = [ ]
  array.delete_if do |e|
    if(yield e)
      dead.push(e)
      true
    else
      false  
    end
  end
  dead
end

a = [1,2,3,4]
x = slice_out(a) { |n| n % 2 == 0 }
# a == [1, 3]
# x == [2, 4]

Answer 3

Array#extract (Rails 6+)

如果您使用的是 Rails，那么从版本 6 开始，有一个方法 Array#extract!，几乎可以满足您的需要。

它删除并返回块返回真值的元素并修改原始数组。

请看下面的例子：

array = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

values = array.extract! { |value| value == 5 }

# array
# => [1, 2, 3, 4, 6, 7, 8, 9, 10]

# values
# => [5]

Answer 4

您可以使用分区。显然这里的块示例并不完全有意义，但会返回已删除的项目和剩余的项目。

a = [1,2,3,4,5,6,7,8,9]
b, a = a.partition { |x| [1,3,5].include?(x) }

b # [1, 3, 5]
a # [2, 4, 6, 7, 8, 9]

Answer 5

这不适用于每个用例，但如果您根据某种条件从数组中一次提取一个项目，您可以这样做：

array = [1,2,3,4,5,6,7,8,9,10]
indexed_array = array.index_by { |a| a }  # or whatever your condition is
item = indexed_array.delete(5)
array = indexed_array.values

Answer 6

你可以使用values_at例如

>> array = [1,2,3,4,5,6,7,8,9,10]
=> [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>> array.values_at(5)
=> [6]