Alex Rivera | Logout

Ruby Select and Reject in one method

Asked 2011-07-21T15:27:21.857
34

Is there any built in method that would combine the functions of Enumerable.select (find all which the block equates to true) and Enumerable.reject (find all which the block equates to false)?

Something like

good, bad = list.magic_method { |obj| obj.good? }
Edit
Report

1 Answer

48

Looks as if Enumerable.partition is exactly what you are after.

= Enumerable.partition

(from ruby core)
------------------------------------------------------------------------------
  enum.partition {| obj | block }  -> [ true_array, false_array ]
  enum.partition                   -> an_enumerator

------------------------------------------------------------------------------

Returns two arrays, the first containing the elements of enum for
which the block evaluates to true, the second containing the rest.

If no block is given, an enumerator is returned instead.

   (1..6).partition {|i| (i&1).zero?}   #=> [[2, 4, 6], [1, 3, 5]]

Interesting, I didn't know that was there. ri is an amazing tool...

answered 2011-07-21T15:32:22.410

Your Answer